Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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>>);
71 ReductionOp<CPUDevice, int32, int32, Eigen::internal::SumReducer<int32>>);
80 ReductionOp<CPUDevice, int32, int64, Eigen::internal::SumReducer<int32>>);
Dreduction_ops_min.cc26 ReductionOp<CPUDevice, type, int32, \
33 ReductionOp<CPUDevice, type, int64, \
47 ReductionOp<GPUDevice, type, int32, \
55 ReductionOp<GPUDevice, type, int64, \
72 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MinReducer<int32>>);
81 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MinReducer<int32>>);
Dreduction_ops_max.cc26 ReductionOp<CPUDevice, type, int32, \
33 ReductionOp<CPUDevice, type, int64, \
47 ReductionOp<GPUDevice, type, int32, \
55 ReductionOp<GPUDevice, type, int64, \
74 ReductionOp<CPUDevice, int32, int32, Eigen::internal::MaxReducer<int32>>);
83 ReductionOp<CPUDevice, int32, int64, Eigen::internal::MaxReducer<int32>>);
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>);
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>>);
Dreduction_ops_euclidean.cc25 ReductionOp<CPUDevice, type, int32, \
31 ReductionOp<CPUDevice, type, int64, \
44 ReductionOp<GPUDevice, type, int32, \
51 ReductionOp<GPUDevice, type, int64, \
Dreduction_ops_prod.cc25 ReductionOp<CPUDevice, type, int32, \
31 ReductionOp<CPUDevice, type, int64, \
44 ReductionOp<GPUDevice, type, int32, \
51 ReductionOp<GPUDevice, type, int64, \
Dreduction_ops_common.h131 class ReductionOp : public OpKernel {
133 explicit ReductionOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc41 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp() function
139 ReductionOp(col_params->merge_op->type_string(), &reduction_op); in Enqueue()
/external/llvm-project/clang/lib/CodeGen/
DCGOpenMPRuntime.h141 const Expr *ReductionOp = nullptr; member
143 const Expr *ReductionOp) in ReductionData()
144 : Shared(Shared), Ref(Ref), Private(Private), ReductionOp(ReductionOp) { in ReductionData()
1389 const Expr *ReductionOp,
DCGOpenMPRuntime.cpp601 getReductionInit(const Expr *ReductionOp) { in getReductionInit() argument
602 if (const auto *CE = dyn_cast<CallExpr>(ReductionOp)) in getReductionInit()
775 EmitDeclareReductionInit ? ClausesData[N].ReductionOp in emitAggregateInitialization()
877 getReductionInit(ClausesData[N].ReductionOp); in emitInitialization()
893 emitInitWithReductionInitializer(CGF, DRD, ClausesData[N].ReductionOp, in emitInitialization()
1021 getReductionInit(ClausesData[N].ReductionOp); in usesReductionInitializer()
5351 const Expr *ReductionOp) { in emitReductionCombiner() argument
5352 if (const auto *CE = dyn_cast<CallExpr>(ReductionOp)) in emitReductionCombiner()
5362 CGF.EmitIgnoredExpr(ReductionOp); in emitReductionCombiner()
5365 CGF.EmitIgnoredExpr(ReductionOp); in emitReductionCombiner()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc570 template <typename ReductionOp>
579 ReductionOp reduce_op = dyn_cast_or_null<ReductionOp>( in MatchBinaryReduceFunction()
/external/llvm-project/clang/lib/Sema/
DSemaOpenMP.cpp127 llvm::PointerUnion<const Expr *, BOKPtrType> ReductionOp; member
131 ReductionOp = BO; in set()
135 ReductionOp = RefExpr; in set()
1488 if (!ReductionData.ReductionOp || in getTopMostTaskgroupReductionData()
1489 ReductionData.ReductionOp.is<const Expr *>()) in getTopMostTaskgroupReductionData()
1492 BOK = ReductionData.ReductionOp.get<ReductionData::BOKPtrType>(); in getTopMostTaskgroupReductionData()
1515 if (!ReductionData.ReductionOp || in getTopMostTaskgroupReductionData()
1516 !ReductionData.ReductionOp.is<const Expr *>()) in getTopMostTaskgroupReductionData()
1519 ReductionRef = ReductionData.ReductionOp.get<const Expr *>(); in getTopMostTaskgroupReductionData()
14853 void push(Expr *Item, Expr *ReductionOp) { in push()
[all …]
/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/llvm-project/mlir/lib/Conversion/VectorToLLVM/
DConvertVectorToLLVM.cpp538 : ConvertToLLVMPattern(vector::ReductionOp::getOperationName(), context, in VectorReductionOpConversion()
545 auto reductionOp = cast<vector::ReductionOp>(op); 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()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp1897 Value reduced = rewriter.create<vector::ReductionOp>( in matchAndRewrite()
2079 return rewriter.create<vector::ReductionOp>(loc, resType, kind, m, in lowerReduction()
DVectorOps.cpp125 static LogicalResult verify(ReductionOp op) { in verify()
177 static void print(OpAsmPrinter &p, ReductionOp op) { in print()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3614 template <typename Derived, typename OpTy, typename ReductionOp,
3668 BuildReduceBody<ReductionOp>(reduce_element_type, &reduction.body(), in matchAndRewrite()
4864 template <typename ConcreteClass, typename OpTy, typename ReductionOp>
4915 BuildReduceBody<ReductionOp>(data_type.getElementType(), in matchAndRewrite()