xla
xla copied to clipboard
openxla
A machine learning compiler for GPUs, CPUs, and ML accelerators
PR #50934: [Go] add bazel rules Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/50934 PR adds Bazel rules to build and test tensorflow/go. Uses [rules_go](https://github.com/bazelbuild/rules_go). Copybara import of the project: -- 84345e113303b95dfba2ba94b083d7fa589e91d2 by...
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "copybara-service[bot] avatar"){kind=avatar}
Fix copy_array_to_devices_with_sharding by making it take a committed argument so that the Array created has the right semantics.
Move PyBuffer methods used by PyArray to c++. ``` def delete(self): ... def unsafe_buffer_pointer(self) -> Any: ... def clone(self) -> ArrayImpl: ... def _copy_single_device_array_to_host_async(self): ... def _single_device_array_to_np_array(self) -> np.ndarray: ......
Update Eigen to commit:2419632cf553ae73d2324f0e8044526cd45cf278 CHANGELOG ========= 2419632cf - Revert change to allFinite(), since the new version does not work for complex numbers. b1beba8a3 - Fix LinAlgSVD example code 7bf2968fe -...
[PJRT:C] Add CopyToDeviceStream to PJRT C API. This is part of what's needed to make send/recv host callbacks work.
Run mobilenet_v2 HLO benchmark on CPU in A/B diff script. Later we will run this twice to make comparisons and extend our benchmark suite to other models. For now we...
Updates Bazel version from 5.3.0 to 6.0.0
Add `-mac_excluded` to TF build/test tag filters Currently, `no_mac` or `nomac` is used to exclude a test from running in the mac environment. However, it is difficult to distinguish between...
Add canonical `singleprint` method to c++/python to easily and uniquely identify a SavedModel.