| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 127 def _normalize_value_destination_pairs(value_destination_pairs): argument
131 value_destination_pairs = list(value_destination_pairs)
133 if not isinstance(value_destination_pairs, (list, tuple)):
135 for pair in value_destination_pairs:
148 def _validate_value_destination_pairs(value_destination_pairs): argument
151 if not value_destination_pairs: return False
152 if not isinstance(value_destination_pairs, (list, tuple)): return False
153 if not all(isinstance(pair, tuple) for pair in value_destination_pairs):
156 for v in value_destination_pairs):
176 def _all_devices_match(value_destination_pairs): argument
[all …]
|
| D | parameter_server_strategy.py | 524 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
525 for _, destinations in value_destination_pairs:
528 value_destination_pairs, options)
|
| D | distribute_lib.py | 2312 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): argument
2382 return self._batch_reduce_to(reduce_op, value_destination_pairs, options)
2384 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
2387 for t, v in value_destination_pairs
2461 def _batch_gather_to(self, value_destination_pairs, axis, options=None): argument
2467 for t, v in value_destination_pairs
|
| D | mirrored_strategy.py | 743 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
745 for value, _ in value_destination_pairs:
753 value_destination_pairs,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 11 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 23 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 11 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 40 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 559 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
563 value_destination_pairs,
566 return test_object.batch_reduce(reduce_op, value_destination_pairs,
|