Searched defs:new_tile (Results 1 – 3 of 3) sorted by relevance
1213 auto new_tile = transposed.tile_assignment(); in PartiallyReplicateTiledShardingOnDims() local1264 auto new_tile = sharding.tile_assignment(); in RemoveShapeDimensions() local
2240 auto new_tile = rhs.sharding().tile_assignment(); in GetDotGroupPartitionContractingLhsRhsShardings() local2249 auto new_tile = lhs.sharding().tile_assignment(); in GetDotGroupPartitionContractingLhsRhsShardings() local
2780 NodeDef* new_tile = AddEmptyNode(new_tile_name); in TrySimplify() local2868 void CopyTileWithInput(const NodeDef* tile, NodeDef* new_tile, in CopyTileWithInput()