Searched refs:reshaped_matching_tiling (Results 1 – 1 of 1) sorted by relevance
1899 Array<int64> reshaped_matching_tiling = matching_sharding.tile_assignment(); in GetNonContractingPartitionGroupedShardingForMatchedOperand() local1900 reshaped_matching_tiling.Reshape(matching_sharding_dims); in GetNonContractingPartitionGroupedShardingForMatchedOperand()1904 ? HloSharding::PartialTile(reshaped_matching_tiling) in GetNonContractingPartitionGroupedShardingForMatchedOperand()1905 : HloSharding::Tile(reshaped_matching_tiling), in GetNonContractingPartitionGroupedShardingForMatchedOperand()