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