Searched refs:unpack_small_tensors (Results 1 – 2 of 2) sorted by relevance
613 def unpack_small_tensors(replica_grads, packing): function
604 return cross_device_utils.unpack_small_tensors(summed_device_grad_packs,