Searched defs:new_tile (Results 1 – 2 of 2) sorted by relevance
112 Array<int64> new_tile(new_tile_dims); in MergeSharding() local1168 auto new_tile = transposed.tile_assignment(); in PartiallyReplicateTiledShardingOnDims() local1199 auto new_tile = sharding.tile_assignment(); in RemoveShapeDimensions() local
2115 auto new_tile = rhs.sharding().tile_assignment(); in PartitionDotGroupOnContracting() local2124 auto new_tile = lhs.sharding().tile_assignment(); in PartitionDotGroupOnContracting() local