| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 72 def reduce_non_distributed_value(reduce_op, device_map, value, destinations): argument
205 reduce_op): argument
231 def reduce(self, reduce_op, per_replica_value, destinations): argument
257 def batch_reduce(self, reduce_op, value_destination_pairs): argument
303 def reduce_implementation(self, reduce_op, per_replica_value, destinations): argument
326 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): argument
382 def reduce_implementation(self, reduce_op, per_replica_value, destinations): argument
395 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): argument
432 reduce_op, argument
667 def reduce_implementation(self, reduce_op, per_replica_value, destinations): argument
[all …]
|
| D | distribute_lib.py | 501 def reduce(self, reduce_op, value): argument
1133 def _reduce(self, reduce_op, value): argument
1139 def reduce_to(self, reduce_op, value, destinations): argument
1161 def _reduce_to(self, reduce_op, value, destinations): argument
1164 def batch_reduce_to(self, reduce_op, value_destination_pairs): argument
1180 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
1475 def all_reduce(self, reduce_op, value): argument
1605 def _reduce_to(self, reduce_op, value, destinations): argument
|
| D | parameter_server_strategy.py | 380 def _reduce_to(self, reduce_op, value, destinations): argument
389 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
|
| D | collective_all_reduce_strategy.py | 430 def _reduce_to(self, reduce_op, value, destinations): argument
|
| D | mirrored_strategy.py | 694 def _reduce_to(self, reduce_op, value, destinations): argument
709 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
|
| D | one_device_strategy.py | 146 def _reduce_to(self, reduce_op, value, destinations): argument
|
| D | input_lib.py | 599 def set_last_step_output(self, name, output, reduce_op=None): argument
|
| D | tpu_strategy.py | 441 def _reduce_to(self, reduce_op, value, destinations): argument
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 323 def testIndexedSlicesAllReduce(self, cross_device_ops_instance, reduce_op, argument
548 def _reduce(test_object, reduce_op, per_replica, destinations): argument
560 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 399 def _ragged_reduce_aggregate(reduce_op, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 165 XlaComputation reduce_op; in RunR2ToR1PredTest() local
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 991 def make_reduce_tests(reduce_op, argument
|
| /external/python/cpython3/Modules/ |
| D | _pickle.c | 3661 const char reduce_op = REDUCE; in save_reduce() local
|