xla
xla copied to clipboard
openxla
A machine learning compiler for GPUs, CPUs, and ML accelerators
follow-up to cl/515714382. removes legacy trace events from the TSL Trace proto.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "copybara-service[bot] avatar"){kind=avatar}
Vendor XLA/TSL into TensorFlow as Bazel dependencies - Test
Improve the accuracy of HermitianEigenDecomposition2x2. Previously there was a threshold on "t" when computing the components of the eigenvector to deal with the case where a previously computed quantity (tau)...
Remove temporarilly the tsl::SourceLocation feature as we will be using absl::Status next.
Vendor XLA/TSL into TensorFlow as Bazel dependencies
Add support for StableHLO Serialized Portable Artifacts in JAX2TF.
Fixes a build failure in XLA on macOS ARM64, which occurred because the presently used setting "@bazel_tools//src/conditions:darwin" collides with the explicitly exported `macos_arm64` setting in case of a macOS ARM64...
Allow specifying index type in XLA TopK. This is one step in the path of updating the corresponding TF/XLA ops to allow different index types.
PR #60038: Remove not needed build file Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/60038 The dependency `mkl_dnn` seems not to be used anywhere. Furthermore, the referenced build file `mkldnn.BUILD` seems not to...