Searched defs:value_destination_pairs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 132 def _normalize_value_destination_pairs(value_destination_pairs): argument
153 def _validate_value_destination_pairs(value_destination_pairs): argument
188 def _all_devices_match(value_destination_pairs, canonicalize_devices=True): argument
400 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): argument
501 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
640 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
871 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
1254 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
|
| D | parameter_server_strategy.py | 529 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| D | distribute_lib.py | 2382 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): argument
2454 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
2544 def _batch_gather_to(self, value_destination_pairs, axis, options=None): argument
|
| D | mirrored_strategy.py | 751 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 559 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
|