Searched refs:copy_sharding (Results 1 – 2 of 2) sorted by relevance
112 slot = xla_sharding.copy_sharding(primary, slot, use_sharding_op=False)
234 def copy_sharding(from_tensor, to_tensor, use_sharding_op=False): function