Home
last modified time | relevance | path

Searched refs:topk_body_fn (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc100 auto topk_body_fn = in TopKWithPartitions() local
168 ForEachIndex(num_partitions - 1, S32, topk_body_fn, in TopKWithPartitions()