Searched refs:num_packs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 467 def __init__(self, num_packs=1): argument
477 if num_packs <= 0:
479 self.num_packs = num_packs
506 num_splits = self.num_packs
609 num_packs=0, argument
613 if num_packs > 0:
614 tensor_packer = _ConcatAndSplitPacker(num_packs)
638 num_packs=1, argument
661 self._num_packs = num_packs
767 def __init__(self, num_packs=1): argument
[all …]
|
| /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\'], "
|
| /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\'], "
|