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