| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 81 def reduce_non_distributed_value(reduce_op, argument
217 reduce_op): argument
272 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
397 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): argument
466 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
498 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
552 def _all_reduce(self, reduce_op, value, replica_id, options): argument
608 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
637 def batch_reduce_implementation(self, reduce_op, value_destination_pairs, argument
676 reduce_op, argument
[all …]
|
| D | cross_device_ops_test.py | 326 reduce_op, prefer_unique_instance_key): argument
373 reduce_op, prefer_unique_instance_key): argument
458 reduce_op, prefer_unique_instance_key): argument
506 reduce_op, prefer_unique_instance_key): argument
635 reduce_op): argument
703 reduce_op): argument
782 implementation, reduce_op): argument
|
| D | parameter_server_strategy_v2.py | 921 def _reduce(self, reduce_op, value): argument
929 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | central_storage_strategy.py | 145 def reduce(self, reduce_op, value, axis): # pylint: disable=useless-super-delegation argument
|
| D | distribute_lib.py | 1317 def reduce(self, reduce_op, value, axis): argument
2015 def reduce(self, reduce_op, value, axis=None): argument
2296 def _reduce(self, reduce_op, value): argument
2301 def reduce_to(self, reduce_op, value, destinations, options=None): argument
2386 def _reduce_to(self, reduce_op, value, destinations, options): argument
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
2467 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
3174 def all_reduce(self, reduce_op, value, options=None): argument
3694 def _reduce_to(self, reduce_op, value, destinations, options): argument
|
| D | mirrored_strategy.py | 730 def _reduce_to(self, reduce_op, value, destinations, options): argument
757 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
785 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
|
| D | collective_all_reduce_strategy.py | 807 def _reduce_to(self, reduce_op, value, destinations, options): argument
833 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
|
| D | one_device_strategy.py | 188 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 | 518 def _reduce_to(self, reduce_op, value, destinations, options): argument
527 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | dtensor_mixed_precision_reduce.cc | 40 mlir::LogicalResult GetAllReduceGroupSize(ReduceOpType reduce_op, in GetAllReduceGroupSize()
65 mlir::LogicalResult MaybeUpcastForReduction(ReduceOpType reduce_op, in MaybeUpcastForReduction()
123 mlir::WalkResult walk_result = function.walk([&](ReduceOpType reduce_op) { in TryMixedPrecisionReduce()
|
| D | collectives.cc | 287 mlir::Operation* input, absl::string_view reduce_op) { in EmitAllReduce()
|
| /external/rust/crates/rayon/src/iter/ |
| D | try_reduce_with.rs | 23 reduce_op: &'r R, field
81 let reduce_op = self.reduce_op; in reduce() localVariable
93 reduce_op: &'r R, field
106 let reduce_op = self.reduce_op; in consume() localVariable
|
| D | try_reduce.rs | 26 reduce_op: &'r R, field
94 reduce_op: &'r R, field
107 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 | 234 reduce_op, batch_reduce): argument
396 def testIndexedSlicesAllReduce(self, cross_device_ops_instance, reduce_op, argument
550 def _reduce(test_object, reduce_op, per_replica, destinations): argument
558 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | reduce_spmd_expander.cc | 99 auto reduce_op = llvm::cast<OpType>(op); in ExtractDims() local
211 mlir::Operation* reduce_op; in ExpandOp() local
|
| D | softmax_spmd_expander.cc | 42 absl::string_view reduce_op, bool keep_dims) { in ComputeGlobalReduce()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | reduce.py | 23 def make_reduce_tests(reduce_op, argument
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reduce_tester.cc | 38 void ReduceTester::Test(tflite::BuiltinOperator reduce_op, in Test()
|
| D | quantized_reduce_tester.cc | 72 void QuantizedReduceTester::Test(tflite::BuiltinOperator reduce_op, in Test()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 49 ncclRedOp_t reduce_op = ToNcclReduction(reduction_kind); in RunAllReduce() local
377 ncclRedOp_t reduce_op = ToNcclReduction(reduction_kind); in RunReduceScatter() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_prim_helpers.h | 213 OpKernelContext* context, int num_segments, ReduceOp reduce_op, in GpuSegmentedReduce()
|
| D | segment_reduction_ops_gpu.cu.h | 57 #define DEFINE_REDUCE_UPDATE_OP_FOR(reduce_op, atomic, nonatomic) \ argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_collective.cc | 213 StringRef reduce_op = all_reduce.reduce_op(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1802 ReductionOp reduce_op = dyn_cast_or_null<ReductionOp>( in MatchBinaryReduceFunction() local
1830 LogicalResult rewriteNonMatchInitValue(mhlo::ReduceOp reduce_op, Value input, in rewriteNonMatchInitValue()
1846 mhlo::ReduceOp reduce_op, Value input, ConstOp reduction_indices, in rewriteNonMatchInitValue()
1853 mhlo::ReduceOp reduce_op, Value input, ConstOp reduction_indices, in rewriteNonMatchInitValue()
1874 mhlo::ReduceOp reduce_op, OpAdaptor adaptor, in matchAndRewrite()
2042 mhlo::ReduceOp reduce_op, OpAdaptor adaptor, in matchAndRewrite()
|