xla icon indicating copy to clipboard operation
xla copied to clipboard

Published 20 hours ago • verified publisher icon openxla

Allow variable k type and output index type in topk.

Open copybara-service[bot] opened this issue 2 years ago • 0 comments
trafficstars

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 of the type, output_index_type. The change is backward-compatible.

Some changes in XLA were required to support the new API.

copybara-service[bot] avatar Mar 11 '23 00:03 copybara-service[bot]