Searched refs:partial_tile (Results 1 – 1 of 1) sorted by relevance
103 def partial_tile(cls, tile_assignment): member in Sharding321 def partial_tile(tensor, tile_assignment, use_sharding_op=False): function332 return Sharding.partial_tile(tile_assignment).apply_to_tensor(445 return Sharding.partial_tile(tile_assignment)