Searched defs:operand_parallel_dims (Results 1 – 5 of 5) sorted by relevance
35 absl::InlinedVector<int64, 1> operand_parallel_dims; member
973 absl::Span<const int64> operand_parallel_dims; in GatherDataOperandShardingFromOutput() local1307 absl::InlinedVector<int64, 1> operand_parallel_dims; in GetGatherBatchParallelDims() local
965 absl::Span<const int64> operand_parallel_dims; in InferShardingFromOperands() local
336 auto operand_parallel_dims = parallel_dims->operand_parallel_dims; in PartitionIndexParallelDimensions() local
1836 auto& operand_parallel_dims = parallel_dims.operand_parallel_dims; in GatherOperandsShardedAcrossParallelDims() local