Home
last modified time | relevance | path

Searched defs:reduction_ops (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc71 llvm::SmallVectorImpl<mlir::TF::DTensorAllReduceOp>* reduction_ops, in CheckReduceAndSumOptimizationCriteria()
125 llvm::SmallVector<mlir::TF::DTensorAllReduceOp, 4> reduction_ops; in OptimizeAllReduceAndSum() local
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmath_ops.py215 def reduction_ops(self): member in ReductionOpsModule
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEReduceMean.cpp46 const unsigned int reduction_ops = reduction_axis.num_dimensions(); in validate_config() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLReduceMean.cpp50 const unsigned int reduction_ops = reduction_axis.num_dimensions(); in validate_config() local
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/clang/AST/
DOpenMPClause.h3468 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3473 helper_expr_range reduction_ops() { in reduction_ops() function
3732 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3737 helper_expr_range reduction_ops() { in reduction_ops() function
3977 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3982 helper_expr_range reduction_ops() { in reduction_ops() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/clang/AST/
DOpenMPClause.h3428 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3433 helper_expr_range reduction_ops() { in reduction_ops() function
3692 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3697 helper_expr_range reduction_ops() { in reduction_ops() function
3937 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3942 helper_expr_range reduction_ops() { in reduction_ops() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/clang/AST/
DOpenMPClause.h3468 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3473 helper_expr_range reduction_ops() { in reduction_ops() function
3732 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3737 helper_expr_range reduction_ops() { in reduction_ops() function
3977 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3982 helper_expr_range reduction_ops() { in reduction_ops() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/clang/AST/
DOpenMPClause.h3468 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3473 helper_expr_range reduction_ops() { in reduction_ops() function
3732 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3737 helper_expr_range reduction_ops() { in reduction_ops() function
3977 helper_expr_const_range reduction_ops() const { in reduction_ops() function
3982 helper_expr_range reduction_ops() { in reduction_ops() function
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h53 const int reduction_ops = reduction_axis.num_dimensions(); in calculate_reduce_mean_shape() local
/aosp_15_r20/external/clang/include/clang/AST/
H A DOpenMPClause.h1850 helper_expr_const_range reduction_ops() const { in reduction_ops() function
1854 helper_expr_range reduction_ops() { in reduction_ops() function
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py24420 reduction_ops = [op for op in ops_and_refs if isinstance(op, ReductionOpInfo)] variable