Searched defs:new_tile (Results 1 – 5 of 5) sorted by relevance
1285 auto new_tile = in PartiallyReplicateTiledShardingOnDims() local1365 auto new_tile = sharding.tile_assignment(); in RemoveShapeDimensions() local
444 auto new_tile = in ExpandOp() local
2557 auto new_tile = rhs.sharding().tile_assignment(); in GetDotGroupPartitionContractingLhsRhsShardings() local2566 auto new_tile = lhs.sharding().tile_assignment(); in GetDotGroupPartitionContractingLhsRhsShardings() local
2779 NodeDef* new_tile = AddEmptyNode(new_tile_name); in TrySimplify() local2867 void CopyTileWithInput(const NodeDef* tile, NodeDef* new_tile, in CopyTileWithInput()
5029 Array<int64_t> new_tile(new_tile_shape); in GetManualSharding() local