| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 117 def _normalize_value_destination_pairs(value_destination_pairs): argument
121 value_destination_pairs = list(value_destination_pairs)
123 if not isinstance(value_destination_pairs, (list, tuple)):
125 for pair in value_destination_pairs:
138 def _validate_value_destination_pairs(value_destination_pairs): argument
141 if not value_destination_pairs: return False
142 if not isinstance(value_destination_pairs, (list, tuple)): return False
143 if not all(isinstance(pair, tuple) for pair in value_destination_pairs):
146 for v in value_destination_pairs):
179 def _all_devices_match(value_destination_pairs): argument
[all …]
|
| D | parameter_server_strategy.py | 389 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
390 for _, destinations in value_destination_pairs:
393 value_destination_pairs)
|
| D | distribute_lib.py | 1164 def batch_reduce_to(self, reduce_op, value_destination_pairs): argument
1178 return self._batch_reduce_to(reduce_op, value_destination_pairs)
1180 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
1183 for t, v in value_destination_pairs
|
| D | mirrored_strategy.py | 709 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
711 reduce_op, 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\'], varargs=None, keywords=None…
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 39 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-cross-device-ops.pbtxt | 11 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-nccl-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-reduction-to-one-device.pbtxt | 12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt | 13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 39 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 560 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
564 value_destination_pairs)
566 return test_object.batch_reduce(reduce_op, value_destination_pairs)
|