Searched refs:indices_parallel_dims_ordered_as_operand (Results 1 – 1 of 1) sorted by relevance
1847 absl::InlinedVector<int64, 1> indices_parallel_dims_ordered_as_operand; in GatherOperandsShardedAcrossParallelDims() local1850 indices_parallel_dims_ordered_as_operand.push_back(idx); in GatherOperandsShardedAcrossParallelDims()1856 indices_parallel_dims_ordered_as_operand, in GatherOperandsShardedAcrossParallelDims()1865 indices_parallel_dims_ordered_as_operand)}; in GatherOperandsShardedAcrossParallelDims()1872 auto target_dims = indices_parallel_dims_ordered_as_operand; in GatherOperandsShardedAcrossParallelDims()1920 for (int i = 0; i < indices_parallel_dims_ordered_as_operand.size(); ++i) { in GatherOperandsShardedAcrossParallelDims()1923 indices_parallel_dims_ordered_as_operand[i]); in GatherOperandsShardedAcrossParallelDims()1932 indices_parallel_dims_ordered_as_operand); in GatherOperandsShardedAcrossParallelDims()