Searched defs:partitioned_contracting_dims (Results 1 – 1 of 1) sorted by relevance
2227 absl::Span<const DotConvDimsMapping::DimsMapping> in GetDotGroupPartitionContractingLhsRhsShardings()2260 absl::Span<const DotConvDimsMapping::DimsMapping> in PartitionDotGroupOnContracting()