Searched refs:__xla_cpu_runtime_TopKF32 (Results 1 – 2 of 2) sorted by relevance
25 extern void __xla_cpu_runtime_TopKF32(tensorflow::int64 batch_size,
71 TF_ATTRIBUTE_NO_SANITIZE_MEMORY void __xla_cpu_runtime_TopKF32( in __xla_cpu_runtime_TopKF32() function