Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dtopk_op.cc190 for (auto top_k_it = top_k->begin(); top_k_it != top_k->end(); in Compute() local
191 ++top_k_it, ++i) { in Compute()
192 indices(b, i) = *top_k_it; in Compute()
195 for (auto top_k_it = filter.unsorted_begin(); in Compute() local
196 top_k_it != filter.unsorted_end(); ++top_k_it, ++i) { in Compute()
197 indices(b, i) = *top_k_it; in Compute()