Searched defs:matching_dims (Results 1 – 1 of 1) sorted by relevance
1886 std::vector<int64> matching_dims; in GetNonContractingPartitionGroupedShardingForMatchedOperand() local2662 std::vector<DotConvDimsMapping::DimsMapping> matching_dims; in PartitionDot() local2689 std::vector<DotConvDimsMapping::DimsMapping> matching_dims; in PartitionDot() local2733 std::vector<DotConvDimsMapping::DimsMapping> matching_dims; in PartitionDot() local