Searched refs:bytes_per_pack (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_util.py | 91 bytes_per_pack=0, argument
125 bytes_per_pack=0, argument
128 if bytes_per_pack < 0:
135 self.bytes_per_pack = bytes_per_pack
156 if options.bytes_per_pack != 0:
157 merged.bytes_per_pack = options.bytes_per_pack
207 def __new__(cls, bytes_per_pack=0, timeout_seconds=None): argument
209 bytes_per_pack=bytes_per_pack, timeout_seconds=timeout_seconds)
211 def __init__(self, bytes_per_pack=0, timeout_seconds=None): argument
|
| D | collective_util_test.py | 28 options = collective_util._OptionsExported(bytes_per_pack=1)
30 self.assertEqual(options.bytes_per_pack, 1)
32 collective_util._OptionsExported(bytes_per_pack=-1)
39 self.assertEqual(options.bytes_per_pack, 50)
|
| D | cross_device_utils_test.py | 131 packs = cross_device_utils.group_by_size(values, bytes_per_pack=200)
145 packs = cross_device_utils.group_by_size(values, bytes_per_pack=0)
160 packs = cross_device_utils.group_by_size(values, bytes_per_pack=1)
|
| D | cross_device_utils.py | 658 def group_by_size(input_tensors, bytes_per_pack): argument
674 if bytes_per_pack == 0:
690 if not packs or last_pack_size > bytes_per_pack:
|
| D | cross_device_ops.py | 1112 options.bytes_per_pack)
1232 options.bytes_per_pack)
1244 values_by_device[i], options.bytes_per_pack)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-collective-hints.pbtxt | 7 …argspec: "args=[\'self\', \'bytes_per_pack\', \'timeout_seconds\'], varargs=None, keywords=None, d…
|
| D | tensorflow.distribute.experimental.-communication-options.pbtxt | 7 …argspec: "args=[\'self\', \'bytes_per_pack\', \'timeout_seconds\', \'implementation\'], varargs=No…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.-collective-hints.pbtxt | 7 …argspec: "args=[\'self\', \'bytes_per_pack\', \'timeout_seconds\'], varargs=None, keywords=None, d…
|
| D | tensorflow.distribute.experimental.-communication-options.pbtxt | 7 …argspec: "args=[\'self\', \'bytes_per_pack\', \'timeout_seconds\', \'implementation\'], varargs=No…
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 728 bytes_per_pack=[0, 1, 4]))
730 bytes_per_pack): argument
731 hints = collective_util.Hints(bytes_per_pack=bytes_per_pack)
|