Searched refs:num_packs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 58 def _mirrored_cross_device_ops(all_reduce_alg, num_packs): argument
83 return cross_device_ops_class(num_packs=num_packs)
89 num_packs=1): argument
135 cross_device_ops=_mirrored_cross_device_ops(all_reduce_alg, num_packs))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 692 def __init__(self, num_packs=1): argument
702 if num_packs <= 0:
704 self.num_packs = num_packs
731 num_splits = self.num_packs
793 def _pack_tensors(device_grads, num_packs=0): argument
795 if num_packs > 0:
796 tensor_packer = _ConcatAndSplitPacker(num_packs)
822 def __init__(self, all_reduce_alg="nccl", num_packs=1): argument
832 self._num_packs = num_packs
954 def __init__(self, num_packs=1): argument
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 9 argspec: "args=[\'self\', \'num_packs\'], varargs=None, keywords=None, defaults=[\'1\'], "
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 9 argspec: "args=[\'self\', \'num_packs\'], varargs=None, keywords=None, defaults=[\'1\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 9 argspec: "args=[\'self\', \'num_packs\'], varargs=None, keywords=None, defaults=[\'1\'], "
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 9 argspec: "args=[\'self\', \'num_packs\'], varargs=None, keywords=None, defaults=[\'1\'], "
|