Searched defs: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): function