Searched defs:new_tile_assignment (Results 1 – 4 of 4) sorted by relevance
462 Array<int64_t> new_tile_assignment = sharding.tile_assignment(); in ReshapeSharding() local499 Array<int64_t> new_tile_assignment(sharding.tile_assignment().dimensions()); in ReverseSharding() local613 Array<int64_t> new_tile_assignment = index_sharding.tile_assignment(); in GatherOutputSharding() local668 Array<int64_t> new_tile_assignment = in GatherIndexSharding() local763 auto new_tile_assignment = relevant_data_sharding.tile_assignment(); in ScatterIndexSharding() local804 Array<int64_t> new_tile_assignment = in ScatterDataSharding() local1850 auto new_tile_assignment = sharding.tile_assignment(); in SplitShardingDimension() local1874 auto new_tile_assignment = transposed_sharding.tile_assignment(); in MergeShardingDimension() local
1370 auto new_tile_assignment = in GetShardingFromUser() local1941 Array<int64_t> new_tile_assignment = op->sharding().tile_assignment(); in InferShardingFromOperands() local
71 auto new_tile_assignment = Array<int64_t>(new_tile_dims); in PartialTile() local
1781 auto new_tile_assignment = to_adjust->tile_assignment(); in GatherOperandsShardedAcrossParallelDims() local