Searched refs:indices_parallel_dims_ordered_as_operand (Results 1 – 1 of 1) sorted by relevance
1721 absl::InlinedVector<int64_t, 1> indices_parallel_dims_ordered_as_operand; in GatherOperandsShardedAcrossParallelDims() local1724 indices_parallel_dims_ordered_as_operand.push_back(idx); in GatherOperandsShardedAcrossParallelDims()1730 indices_parallel_dims_ordered_as_operand, in GatherOperandsShardedAcrossParallelDims()1739 indices_parallel_dims_ordered_as_operand)}; in GatherOperandsShardedAcrossParallelDims()1746 auto target_dims = indices_parallel_dims_ordered_as_operand; in GatherOperandsShardedAcrossParallelDims()1795 for (int i = 0; i < indices_parallel_dims_ordered_as_operand.size(); ++i) { in GatherOperandsShardedAcrossParallelDims()1798 indices_parallel_dims_ordered_as_operand[i]); in GatherOperandsShardedAcrossParallelDims()1807 indices_parallel_dims_ordered_as_operand); in GatherOperandsShardedAcrossParallelDims()