Searched refs:out_indices_batch (Results 1 – 1 of 1) sorted by relevance
63 tensorflow::int32* out_indices_batch = out_indices + batch * k; in TopK() local64 std::copy(temp_indices.begin(), kth_element, out_indices_batch); in TopK()