xla
xla copied to clipboard
Published
20 hours ago •
openxla
openxla
Allow specifying index type in XLA TopK.
trafficstars
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.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "Published by a pub.dev verified publisher"){kind=small}