Searched defs:use_sharding_op (Results 1 – 1 of 1) sorted by relevance
208 use_sharding_op=False, argument281 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): argument308 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): argument318 use_sharding_op=False): argument329 use_sharding_op=False, argument352 use_sharding_op=False, argument373 use_sharding_op=False, argument561 use_sharding_op=False, argument