Searched refs:DimsMapping (Results 1 – 2 of 2) sorted by relevance
386 struct DimsMapping { struct393 std::vector<DimsMapping> batch_dims;394 std::vector<DimsMapping> contracting_dims;395 std::vector<DimsMapping> lhs_non_contracting_dims;396 std::vector<DimsMapping> rhs_non_contracting_dims;397 std::vector<DimsMapping> conv_spatial_dims;
349 [&](const DotConvDimsMapping::DimsMapping& mapping) { in ComputeDimensionIndexMapping()1883 absl::Span<const DotConvDimsMapping::DimsMapping> partitioned_dims) { in GetNonContractingPartitionGroupedShardingForMatchedOperand()1917 absl::Span<const DotConvDimsMapping::DimsMapping> matching_partitioned_dims, in GetNonContractingPartitionGroupedShardingForOtherOperand()1918 absl::Span<const DotConvDimsMapping::DimsMapping> in GetNonContractingPartitionGroupedShardingForOtherOperand()1920 absl::Span<const DotConvDimsMapping::DimsMapping> other_contracting_dims) { in GetNonContractingPartitionGroupedShardingForOtherOperand()1982 absl::Span<const DotConvDimsMapping::DimsMapping> in PartitionDotGroupOnNonContracting()2077 absl::Span<const DotConvDimsMapping::DimsMapping> in PartitionDotGroupOnContracting()2317 const std::vector<DotConvDimsMapping::DimsMapping>& in LhsIsBestMatchForNonContractingPartitioning()2321 const std::vector<DotConvDimsMapping::DimsMapping>& in LhsIsBestMatchForNonContractingPartitioning()2367 for (const DotConvDimsMapping::DimsMapping& dim : in LhsIsBestMatchForNonContractingPartitioning()[all …]