Searched refs:index_parallel_dim (Results 1 – 1 of 1) sorted by relevance
1314 int index_parallel_dim = index_parallel_in_dim[i]; in GetGatherBatchParallelDims() local1315 if (index_parallel_dim == -1) { in GetGatherBatchParallelDims()1318 if (absl::c_linear_search(indices_parallel_dims, index_parallel_dim)) { in GetGatherBatchParallelDims()1323 indices_parallel_dims.push_back(index_parallel_dim); in GetGatherBatchParallelDims()1367 const int64 index_parallel_dim = parallel_dims.index_parallel_in_dim[i]; in GatherOutputAlignedOperandParallelDims() local1369 if (index_parallel_dim == -1) { in GatherOutputAlignedOperandParallelDims()1376 if (parallel_dims.indices_parallel_dims[j] == index_parallel_dim) { in GatherOutputAlignedOperandParallelDims()