Searched refs:num_indices_from_element (Results 1 – 1 of 1) sorted by relevance
1281 const int64 num_indices_from_element = in GetGatherBatchParallelDims() local1287 for (int j = 0; j < num_indices_from_element; ++j) { in GetGatherBatchParallelDims()1293 concatenated_dims += num_indices_from_element; in GetGatherBatchParallelDims()1298 const int64 num_indices_from_element = in GetGatherBatchParallelDims() local1302 index_parallel_in_dim.assign(num_indices_from_element, in GetGatherBatchParallelDims()