Searched defs:value_destination_pairs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 130 def _normalize_value_destination_pairs(value_destination_pairs): argument
151 def _validate_value_destination_pairs(value_destination_pairs): argument
186 def _all_devices_match(value_destination_pairs, canonicalize_devices=True): argument
397 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): argument
498 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
637 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
868 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 | 527 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| D | distribute_lib.py | 2389 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): argument
2461 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
2551 def _batch_gather_to(self, value_destination_pairs, axis, options=None): argument
|
| D | mirrored_strategy.py | 757 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 558 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
|