Searched defs:target_dims (Results 1 – 1 of 1) sorted by relevance
1797 absl::Span<const int64> target_dims, in CreateMatchingShardingOnDims()1872 auto target_dims = indices_parallel_dims_ordered_as_operand; in GatherOperandsShardedAcrossParallelDims() local