Searched refs:matching_dims (Results 1 – 1 of 1) sorted by relevance
1886 std::vector<int64> matching_dims; in GetNonContractingPartitionGroupedShardingForMatchedOperand() local1894 matching_dims.push_back(md); in GetNonContractingPartitionGroupedShardingForMatchedOperand()1906 matching_dims), in GetNonContractingPartitionGroupedShardingForMatchedOperand()2662 std::vector<DotConvDimsMapping::DimsMapping> matching_dims; in PartitionDot() local2667 matching_dims.push_back(dim); in PartitionDot()2670 if (!matching_dims.empty()) { in PartitionDot()2675 matching_dims, rhs_non_contracting_partitions, in PartitionDot()2689 std::vector<DotConvDimsMapping::DimsMapping> matching_dims; in PartitionDot() local2694 matching_dims.push_back(dim); in PartitionDot()2697 if (!matching_dims.empty()) { in PartitionDot()[all …]