Searched refs:values_batch (Results 1 – 1 of 1) sorted by relevance
39 const T* values_batch = values + batch * input_size; in TopK() local54 tensorflow::int32 v1 = convert_to_int(values_batch[i1]); in TopK()55 tensorflow::int32 v2 = convert_to_int(values_batch[i2]); in TopK()66 out_values_batch[i] = values_batch[temp_indices[i]]; in TopK()