Searched defs:num_packs (Results 1 – 1 of 1) sorted by relevance
712 def __init__(self, num_packs=1): argument813 def _pack_tensors(device_grads, num_packs=0): argument842 def __init__(self, all_reduce_alg="nccl", num_packs=1): argument975 def __init__(self, num_packs=1): argument1017 def __init__(self, num_packs=1): argument