| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 84 reduce_op, value, destinations, num_replicas_in_graph): argument
201 reduce_op): argument
255 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
381 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): argument
450 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
482 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
532 def _all_reduce(self, reduce_op, value, replica_id, options): argument
588 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
617 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
656 reduce_op, argument
[all …]
|
| D | cross_device_ops_test.py | 323 reduce_op, prefer_unique_instance_key): argument
365 reduce_op, prefer_unique_instance_key): argument
445 implementation, reduce_op, argument
489 implementation, reduce_op, argument
575 implementation, reduce_op): argument
|
| D | central_storage_strategy.py | 149 def reduce(self, reduce_op, value, axis): # pylint: disable=useless-super-delegation argument
|
| D | distribute_lib.py | 1268 def reduce(self, reduce_op, value, axis): argument
1954 def reduce(self, reduce_op, value, axis=None): argument
2219 def _reduce(self, reduce_op, value): argument
2224 def reduce_to(self, reduce_op, value, destinations, options=None): argument
2309 def _reduce_to(self, reduce_op, value, destinations, options): argument
2312 def batch_reduce_to(self, reduce_op, value_destination_pairs, options=None): argument
2384 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
2390 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
3053 def all_reduce(self, reduce_op, value, options=None): argument
3465 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | parameter_server_strategy_v2.py | 726 def _reduce(self, reduce_op, value): argument
|
| D | collective_all_reduce_strategy.py | 749 def _reduce_to(self, reduce_op, value, destinations, options): argument
775 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
|
| D | one_device_strategy.py | 190 def reduce(self, reduce_op, value, axis): # pylint: disable=useless-super-delegation argument
396 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | parameter_server_strategy.py | 515 def _reduce_to(self, reduce_op, value, destinations, options): argument
524 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| D | mirrored_strategy.py | 719 def _reduce_to(self, reduce_op, value, destinations, options): argument
743 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| D | distribute_lib_test.py | 97 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | tpu_strategy.py | 1215 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | input_lib.py | 2044 def set_last_step_output(self, name, output, reduce_op=None): argument
|
| /external/rust/crates/rayon/src/iter/ |
| D | try_reduce_with.rs | 22 reduce_op: &'r R, field
80 let reduce_op = self.reduce_op; in reduce() localVariable
92 reduce_op: &'r R, field
105 let reduce_op = self.reduce_op; in consume() localVariable
|
| D | try_reduce.rs | 25 reduce_op: &'r R, field
93 reduce_op: &'r R, field
106 let reduce_op = self.reduce_op; in consume() localVariable
|
| D | reduce.rs | 20 reduce_op: &'r R, field
82 reduce_op: &'r R, field
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 237 reduce_op, batch_reduce): argument
399 def testIndexedSlicesAllReduce(self, cross_device_ops_instance, reduce_op, argument
551 def _reduce(test_object, reduce_op, per_replica, destinations): argument
559 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | reduce.py | 27 def make_reduce_tests(reduce_op, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 66 void ConvertToReductionOperator(Location loc, scf::ReduceOp reduce_op, in ConvertToReductionOperator()
194 lmhlo::ReduceOp reduce_op, ArrayRef<Value> /*args*/, in matchAndRewrite()
228 lmhlo::ReduceOp reduce_op, ConversionPatternRewriter* rewriter) const { in CreateReduceOpInNestedParallelLoops()
372 scf::ReduceOp reduce_op = CreateReduceOpInNestedParallelLoops( in matchAndRewrite() local
|
| D | lhlo_legalize_to_gpu.cc | 53 ReduceOp reduce_op, ArrayRef<Value> args, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 125 ncclRedOp_t reduce_op = ToNcclReduction(config_.reduction_kind); in RunNcclCollective() local
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reduce_tester.cc | 37 void ReduceTester::Test(tflite::BuiltinOperator reduce_op, in Test()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 434 def ragged_reduce_aggregate(reduce_op, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 579 ReductionOp reduce_op = dyn_cast_or_null<ReductionOp>( in MatchBinaryReduceFunction() local
612 mhlo::ReduceOp reduce_op, ArrayRef<Value> args, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 496 TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status); in RegisterCollectiveMulFunction() local
|
| /external/mesa3d/src/amd/compiler/ |
| D | aco_lower_to_hw_instr.cpp | 485 void emit_reduction(lower_context *ctx, aco_opcode op, ReduceOp reduce_op, unsigned cluster_size, P… in emit_reduction()
|