Home
last modified time | relevance | path

Searched defs:reduce_op (Results 1 – 25 of 61) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcross_device_ops.py81 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 …]
H A Dcross_device_ops_test.py326 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
H A Ddistribute_lib.py1317 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
H A Dmirrored_strategy.py730 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
H A Dparameter_server_strategy_v2.py921 def _reduce(self, reduce_op, value): argument
929 def _reduce_to(self, reduce_op, value, destinations, options): argument
H A Dcollective_all_reduce_strategy.py807 def _reduce_to(self, reduce_op, value, destinations, options): argument
833 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): argument
H A Done_device_strategy.py188 def reduce(self, reduce_op, value, axis): # pylint: disable=useless-super-delegation argument
396 def _reduce_to(self, reduce_op, value, destinations, options): argument
H A Dparameter_server_strategy.py518 def _reduce_to(self, reduce_op, value, destinations, options): argument
527 def _batch_reduce_to(self, reduce_op, value_destination_pairs, options): argument
H A Dcentral_storage_strategy.py145 def reduce(self, reduce_op, value, axis): # pylint: disable=useless-super-delegation argument
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFunctional.cpp96 c10d::ReduceOp to_reduce_op(const std::string& reduce_op) { in to_reduce_op()
106 std::string reduce_op, in all_reduce_()
121 std::string reduce_op, in all_reduce()
130 std::string reduce_op, in all_reduce_coalesced_()
147 std::string reduce_op, in all_reduce_coalesced()
227 std::string reduce_op, in reduce_scatter_tensor_coalesced()
249 std::string reduce_op, in reduce_scatter_tensor()
463 const std::string& reduce_op, in forward()
510 const std::string& reduce_op, in reduce_scatter_tensor_autograd()
H A DCUDASymmetricMemoryOps.cu111 std::string reduce_op, in multimem_all_reduce_()
192 std::string reduce_op, in multimem_one_shot_all_reduce()
/aosp_15_r20/external/rust/android-crates-io/crates/rayon/src/iter/
Dtry_reduce_with.rs23 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
Dtry_reduce.rs26 reduce_op: &'r R, field
94 reduce_op: &'r R, field
107 let reduce_op = self.reduce_op; in consume() localVariable
Dreduce.rs20 reduce_op: &'r R, field
82 reduce_op: &'r R, field
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_mixed_precision_reduce.cc40 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSpmmReduceKernel.cpp436 ReductionType reduce_op) { in spmm_reduce_kernel()
454 ReductionType reduce_op) { in spmm_reduce_arg_kernel()
472 ReductionType reduce_op) { in spmm_reduce_backward_input_kernel()
490 ReductionType reduce_op) { in spmm_reduce_backward_input_arg_kernel()
521 ReductionType reduce_op) { in spmm_reduce_backward_other_kernel()
546 ReductionType reduce_op) { in spmm_reduce_backward_other_arg_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dcross_device_ops_test.py234 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
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives_impl.py23 def _all_reduce(input, reduce_op, tag, ranks, group_size): argument
32 def _all_reduce_coalesced(inputs, reduce_op, tag, ranks, group_size): argument
H A D_functional_collectives.py848 def _reduce_scatter_tensor_meta(input, reduce_op, tag, rankset, group_size): argument
917 def _reduce_scatter_tensor_native_meta(inp, reduce_op, group_size, group_name): argument
924 inputs, reduce_op, group_size, group_name argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dreduce_spmd_expander.cc99 auto reduce_op = llvm::cast<OpType>(op); in ExtractDims() local
211 mlir::Operation* reduce_op; in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_all_reduce_thunk.cc49 ncclRedOp_t reduce_op = ToNcclReduction(reduction_kind); in RunAllReduce() local
377 ncclRedOp_t reduce_op = ToNcclReduction(reduction_kind); in RunReduceScatter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc1802 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dreduce.py23 def make_reduce_tests(reduce_op, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dreduce_tester.cc38 void ReduceTester::Test(tflite::BuiltinOperator reduce_op, in Test()
H A Dquantized_reduce_tester.cc72 void QuantizedReduceTester::Test(tflite::BuiltinOperator reduce_op, in Test()

123