Searched refs:matching_sharding_dims (Results 1 – 1 of 1) sorted by relevance
1884 std::vector<int64> matching_sharding_dims = in GetNonContractingPartitionGroupedShardingForMatchedOperand() local1892 matching_sharding_dims[md] = in GetNonContractingPartitionGroupedShardingForMatchedOperand()1900 reshaped_matching_tiling.Reshape(matching_sharding_dims); in GetNonContractingPartitionGroupedShardingForMatchedOperand()