Searched defs:num_packs (Results 1 – 2 of 2) sorted by relevance
58 def _mirrored_cross_device_ops(all_reduce_alg, num_packs): argument89 num_packs=1): argument
692 def __init__(self, num_packs=1): argument793 def _pack_tensors(device_grads, num_packs=0): argument822 def __init__(self, all_reduce_alg="nccl", num_packs=1): argument954 def __init__(self, num_packs=1): argument996 def __init__(self, num_packs=1): argument