xla
xla copied to clipboard
openxla
A machine learning compiler for GPUs, CPUs, and ML accelerators
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "copybara-service[bot] avatar"){kind=avatar}
Reverts 17e8964c21d95bb7fd03b44524d23dd1ffa1037c
XLA builds with Bazel at the moment, is it desirable to also have a CMake build? Who would benefit from this and what workflow will this enable that aren't doable...
By default, layout assignment tries to assign a layout to transposes that make them a bitcast. This layout is then propagated inside the HloComputation, which means if it does not...
[mhlo] Pretty printing for all ops with dense array (dimension, permutation, etc)
Allow variable k type and output index type in topk. This is to help support smaller model sizes for some tflite models. Requires expanding the TF API to allow specification...
[XLA:CPU Next] Merge transform_matmul and transform_dot.
Links Android targets in mostly static mode in bazelrc.
Vendor XLA/TSL into TensorFlow as Bazel dependencies