Searched defs:use_sharding_op (Results 1 – 1 of 1) sorted by relevance
167 use_sharding_op=False): argument234 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument261 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): argument271 use_sharding_op=False): argument282 use_sharding_op=False): argument302 use_sharding_op=False, argument321 def partial_tile(tensor, tile_assignment, use_sharding_op=False): argument452 use_sharding_op=False): argument