Searched defs:new_tile_assignment (Results 1 – 4 of 4) sorted by relevance
400 Array<int64> new_tile_assignment = sharding.tile_assignment(); in ReshapeSharding() local418 Array<int64> new_tile_assignment(sharding.tile_assignment().dimensions()); in ReverseSharding() local524 Array<int64> new_tile_assignment = index_sharding.tile_assignment(); in GatherOutputSharding() local566 Array<int64> new_tile_assignment = output_sharding.tile_assignment(); in GatherIndexSharding() local660 Array<int64> new_tile_assignment = data_sharding.tile_assignment(); in ScatterIndexSharding() local693 Array<int64> new_tile_assignment = index_sharding.tile_assignment(); in ScatterDataSharding() local
765 Array<int64> new_tile_assignment = op->sharding().tile_assignment(); in InferShardingFromOperands() local1194 auto new_tile_assignment = in GetShardingFromUser() local
57 auto new_tile_assignment = Array<int64>(new_tile_dims); in PartialTile() local
1906 auto new_tile_assignment = to_adjust->tile_assignment(); in GatherOperandsShardedAcrossParallelDims() local