Home
last modified time | relevance | path

Searched refs:ReductionOp (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops_max.cc26 ReductionOp<CPUDevice, type, int32, Eigen::internal::MaxReducer<type>>); \
32 ReductionOp<CPUDevice, type, int64, Eigen::internal::MaxReducer<type>>);
45 ReductionOp<GPUDevice, type, int32, Eigen::internal::MaxReducer<type>>); \
52 ReductionOp<GPUDevice, type, int64, Eigen::internal::MaxReducer<type>>);
70 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MaxReducer<int32>>);
79 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MaxReducer<int32>>);
92 ReductionOp<SYCLDevice, type, int32, \
99 ReductionOp<SYCLDevice, type, int64, \
112 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MaxReducer<int32>>);
121 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MaxReducer<int32>>);
Dreduction_ops_sum.cc26 ReductionOp<CPUDevice, type, int32, Eigen::internal::SumReducer<type>>); \
32 ReductionOp<CPUDevice, type, int64, Eigen::internal::SumReducer<type>>);
45 ReductionOp<GPUDevice, type, int32, Eigen::internal::SumReducer<type>>); \
52 ReductionOp<GPUDevice, type, int64, Eigen::internal::SumReducer<type>>);
72 ReductionOp<CPUDevice, int32, int32, Eigen::internal::SumReducer<int32>>);
81 ReductionOp<CPUDevice, int32, int64, Eigen::internal::SumReducer<int32>>);
92 ReductionOp<SYCLDevice, type, int32, \
99 ReductionOp<SYCLDevice, type, int64, \
112 ReductionOp<CPUDevice, int32, int32, Eigen::internal::SumReducer<int32>>);
121 ReductionOp<CPUDevice, int32, int64, Eigen::internal::SumReducer<int32>>);
Dreduction_ops_min.cc26 ReductionOp<CPUDevice, type, int32, Eigen::internal::MinReducer<type>>); \
32 ReductionOp<CPUDevice, type, int64, Eigen::internal::MinReducer<type>>);
45 ReductionOp<GPUDevice, type, int32, Eigen::internal::MinReducer<type>>); \
52 ReductionOp<GPUDevice, type, int64, Eigen::internal::MinReducer<type>>);
68 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MinReducer<int32>>);
77 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MinReducer<int32>>);
90 ReductionOp<SYCLDevice, type, int32, \
97 ReductionOp<SYCLDevice, type, int64, \
110 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MinReducer<int32>>);
119 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MinReducer<int32>>);
Dreduction_ops_euclidean.cc25 ReductionOp<CPUDevice, type, int32, \
31 ReductionOp<CPUDevice, type, int64, \
44 ReductionOp<GPUDevice, type, int32, \
51 ReductionOp<GPUDevice, type, int64, \
69 ReductionOp<SYCLDevice, type, int32, \
76 ReductionOp<SYCLDevice, type, int64, \
Dreduction_ops_mean.cc26 ReductionOp<CPUDevice, type, int32, functor::MeanReducer<type>>); \
32 ReductionOp<CPUDevice, type, int64, functor::MeanReducer<type>>);
45 ReductionOp<GPUDevice, type, int32, functor::MeanReducer<type>>); \
52 ReductionOp<GPUDevice, type, int64, functor::MeanReducer<type>>);
70 ReductionOp<SYCLDevice, type, int32, functor::MeanReducer<type>>); \
77 ReductionOp<SYCLDevice, type, int64, functor::MeanReducer<type>>);
Dreduction_ops_prod.cc25 ReductionOp<CPUDevice, type, int32, \
31 ReductionOp<CPUDevice, type, int64, \
44 ReductionOp<GPUDevice, type, int32, \
51 ReductionOp<GPUDevice, type, int64, \
70 ReductionOp<SYCLDevice, type, int32, \
77 ReductionOp<SYCLDevice, type, int64, \
Dreduction_ops_all.cc25 ReductionOp<CPUDevice, bool, int32, Eigen::internal::AndReducer>);
31 ReductionOp<CPUDevice, bool, int64, Eigen::internal::AndReducer>);
39 ReductionOp<GPUDevice, bool, int32, Eigen::internal::AndReducer>);
45 ReductionOp<GPUDevice, bool, int64, Eigen::internal::AndReducer>);
Dreduction_ops_any.cc25 ReductionOp<CPUDevice, bool, int32, Eigen::internal::OrReducer>);
31 ReductionOp<CPUDevice, bool, int64, Eigen::internal::OrReducer>);
39 ReductionOp<GPUDevice, bool, int32, Eigen::internal::OrReducer>);
45 ReductionOp<GPUDevice, bool, int64, Eigen::internal::OrReducer>);
Dcollective_nccl_reducer.cc27 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp() function
50 Status s = ReductionOp(col_params_->merge_op->type_string(), &reduction_op); in Run()
Dreduction_ops_common.h138 class ReductionOp : public OpKernel {
140 explicit ReductionOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
/external/clang/lib/Sema/
DSemaOpenMP.cpp9453 ExprResult ReductionOp; in ActOnOpenMPReductionClause() local
9476 ReductionOp = new (Context) in ActOnOpenMPReductionClause()
9479 ReductionOp = BuildBinOp(DSAStack->getCurScope(), in ActOnOpenMPReductionClause()
9481 if (ReductionOp.isUsable()) { in ActOnOpenMPReductionClause()
9483 ReductionOp = in ActOnOpenMPReductionClause()
9485 BO_Assign, LHSDRE, ReductionOp.get()); in ActOnOpenMPReductionClause()
9488 ReductionOp.get(), SourceLocation(), LHSDRE, SourceLocation(), in ActOnOpenMPReductionClause()
9490 ReductionOp = in ActOnOpenMPReductionClause()
9494 ReductionOp = ActOnFinishFullExpr(ReductionOp.get()); in ActOnOpenMPReductionClause()
9496 if (ReductionOp.isInvalid()) in ActOnOpenMPReductionClause()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1968 template <typename Derived, typename OpTy, typename ReductionOp,
2014 BuildReduceBody<ReductionOp>(reduce_element_type, &reduction.body(), in matchAndRewrite()
3090 template <typename ConcreteClass, typename OpTy, typename ReductionOp>
3141 BuildReduceBody<ReductionOp>(data_type.getElementType(), in matchAndRewrite()
/external/clang/lib/CodeGen/
DCGOpenMPRuntime.cpp4205 const Expr *ReductionOp) { in emitReductionCombiner() argument
4206 if (auto *CE = dyn_cast<CallExpr>(ReductionOp)) in emitReductionCombiner()
4215 CGF.EmitIgnoredExpr(ReductionOp); in emitReductionCombiner()
4218 CGF.EmitIgnoredExpr(ReductionOp); in emitReductionCombiner()
4310 const Expr *ReductionOp, in emitSingleReductionCombiner() argument
4321 emitReductionCombiner(CGF, ReductionOp); in emitSingleReductionCombiner()
4325 emitReductionCombiner(CGF, ReductionOp); in emitSingleReductionCombiner()
DCGStmtOpenMP.cpp387 getReductionInit(const Expr *ReductionOp) { in getReductionInit() argument
388 if (auto *CE = dyn_cast<CallExpr>(ReductionOp)) in getReductionInit()