| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 130 def _normalize_value_destination_pairs(value_destination_pairs): argument
134 value_destination_pairs = list(value_destination_pairs)
136 if not isinstance(value_destination_pairs, (list, tuple)):
138 for pair in value_destination_pairs:
151 def _validate_value_destination_pairs(value_destination_pairs): argument
154 if not value_destination_pairs: return False
155 if not isinstance(value_destination_pairs, (list, tuple)): return False
156 if not all(isinstance(pair, tuple) for pair in value_destination_pairs):
159 for v in value_destination_pairs):
186 def _all_devices_match(value_destination_pairs, canonicalize_devices=True): argument
[all …]
|
| D | parameter_server_strategy.py | 527 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
528 for _, destinations in value_destination_pairs:
531 value_destination_pairs, options)
|
| D | distribute_lib.py | 2389 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): argument
2459 return self._batch_reduce_to(reduce_op, value_destination_pairs, options)
2461 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
2464 for t, v in value_destination_pairs
2551 def _batch_gather_to(self, value_destination_pairs, axis, options=None): argument
2557 for t, v in value_destination_pairs
|
| D | mirrored_strategy.py | 757 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
759 for value, _ in value_destination_pairs:
767 value_destination_pairs,
|
| /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.-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.-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.-strategy-extended.pbtxt | 40 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\', \'options\'], varargs=None, …
|
| /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/python/distribute/v1/ |
| D | cross_device_ops_test.py | 558 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
562 value_destination_pairs,
565 return test_object.batch_reduce(reduce_op, value_destination_pairs,
|