Home
last modified time | relevance | path

Searched refs:ReduceNode (Results 1 – 25 of 53) sorted by relevance

123

/dports/misc/thrill/thrill-12c5b59bca66df93b66628b3829027bd0f110dd9/thrill/api/
H A Dreduce_by_key.hpp64 class ReduceNode final : public DOpNode<ValueType> class
87 explicit Emitter(ReduceNode* node) : node_(node) { } in Emitter()
92 ReduceNode* node_;
101 ReduceNode(const ParentDIA& parent, in ReduceNode() function in thrill::api::ReduceNode
352 using ReduceNode = api::ReduceNode< in ReduceByKey() typedef
357 auto node = tlx::make_counting<ReduceNode>( in ReduceByKey()
451 using ReduceNode = api::ReduceNode< in ReducePair() typedef
457 auto node = tlx::make_counting<ReduceNode>( in ReducePair()
H A Dreduce_to_index.hpp319 using ReduceNode = ReduceToIndexNode< in ReduceToIndex() typedef
323 auto node = tlx::make_counting<ReduceNode>( in ReduceToIndex()
377 using ReduceNode = ReduceToIndexNode< in ReduceToIndex() typedef
381 auto node = tlx::make_counting<ReduceNode>( in ReduceToIndex()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/autodiff/
H A Dad_util.cc49 if (const ReduceNode* red = expr.as<ReduceNode>()) { in CloneReduction()
87 if (const ReduceNode* red = exprs[0].as<ReduceNode>()) { in ComputeOpFromExprs()
104 if (const ReduceNode* red = expr.as<ReduceNode>()) { in TensorFromExpr()
171 if (inline_reductions_ || !op_comp->body[0].as<ReduceNode>()) { in VisitExpr_()
H A Djacobian.cc163 PrimExpr VisitExpr_(const ReduceNode* op) { in VisitExpr_()
182 const ReduceNode* new_op = expr_with_new_axes.as<ReduceNode>(); in VisitExpr_()
344 if (const ReduceNode* red = new_body.as<ReduceNode>()) { in Jacobian()
H A Dad_simplify.cc792 if (const ReduceNode* red = expr.as<ReduceNode>()) { in SimplifyReductionDomain()
939 virtual PrimExpr VisitExpr_(const ReduceNode* op) { in VisitExpr_()
1068 PrimExpr VisitExpr_(const ReduceNode* op) final { in VisitExpr_()
1129 if (const ReduceNode* red = expr.as<ReduceNode>()) { in LiftReductions()
1154 if (const ReduceNode* red = expr.as<ReduceNode>()) { in RemoveJacobianAndLiftNonzeroCondImpl()
1176 red = new_red.as<ReduceNode>(); in RemoveJacobianAndLiftNonzeroCondImpl()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/operation/
H A Dcompute_op.cc57 inline bool ReduceEqual(const tir::ReduceNode* a, const tir::ReduceNode* b) { in ReduceEqual()
138 if (n->body[0]->IsInstance<tir::ReduceNode>()) { in ComputeOp()
139 const tir::ReduceNode* reduce = n->body[0].as<tir::ReduceNode>(); in ComputeOp()
174 if (this->body[0]->IsInstance<tir::ReduceNode>()) { in ReplaceInputs()
179 const tir::ReduceNode* r = new_reduce.as<tir::ReduceNode>(); in ReplaceInputs()
181 auto n = make_object<tir::ReduceNode>(*r); in ReplaceInputs()
301 const ReduceNode* reduce = op->body[0].as<ReduceNode>(); in MakeReduction()
512 : compute_(compute), reduce_(compute->body[0].as<tir::ReduceNode>()) {} in ComputeVerifier()
524 const tir::ReduceNode* reduce = e.as<tir::ReduceNode>(); in Run()
547 void VisitExpr_(const tir::ReduceNode* op) final { in VisitExpr_()
[all …]
H A Dcross_thread_reduction.cc96 std::vector<const ReduceNode*> reduces(size); in MakeCrossThreadReduction()
98 const ReduceNode* reduce = self->body[i].as<ReduceNode>(); in MakeCrossThreadReduction()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/
H A Dschedule_dataflow_rewrite.cc73 const tir::ReduceNode* new_reduce = new_e.as<tir::ReduceNode>(); in VisitExpr_()
88 using tir::ReduceNode; in InjectPredicate()
91 const ReduceNode* reduce = body.as<ReduceNode>(); in InjectPredicate()
95 auto n = make_object<ReduceNode>(*reduce); in InjectPredicate()
124 inline bool ReduceEqual(const tir::ReduceNode* a, const tir::ReduceNode* b) { in ReduceEqual()
301 const tir::ReduceNode* reduce_body = body.as<tir::ReduceNode>(); in CacheWriteWithReLayout()
530 const tir::ReduceNode* reduce = new_body[j][0].as<tir::ReduceNode>(); in InjectInline()
532 const tir::ReduceNode* reduce_ = new_body[j][k].as<tir::ReduceNode>(); in InjectInline()
542 const tir::ReduceNode* r = new_value.as<tir::ReduceNode>(); in InjectInline()
703 using tir::ReduceNode; in rfactor()
[all …]
/dports/misc/thrill/thrill-12c5b59bca66df93b66628b3829027bd0f110dd9/tests/api/
H A Dreduce_node_test.cpp83 TEST(ReduceNode, ReduceModulo2CorrectResults) { in TEST() argument
132 TEST(ReduceNode, ReduceModuloPairsCorrectResults) { in TEST() argument
208 TEST(ReduceNode, ReduceToIndexCorrectResults) { in TEST() argument
/dports/lang/v8/v8-9.6.180.12/test/unittests/compiler/
H A Dgraph-reducer-unittest.cc427 void ReduceNode(Node* node, Reducer* r) { in ReduceNode() function in v8::internal::compiler::graph_reducer_unittest::GraphReducerTest
430 reducer.ReduceNode(node); in ReduceNode()
433 void ReduceNode(Node* node, Reducer* r1, Reducer* r2) { in ReduceNode() function in v8::internal::compiler::graph_reducer_unittest::GraphReducerTest
437 reducer.ReduceNode(node); in ReduceNode()
440 void ReduceNode(Node* node, Reducer* r1, Reducer* r2, Reducer* r3) { in ReduceNode() function in v8::internal::compiler::graph_reducer_unittest::GraphReducerTest
445 reducer.ReduceNode(node); in ReduceNode()
485 ReduceNode(node1, &r); in TEST_F()
497 ReduceNode(node0, &r1, &r2); in TEST_F()
511 ReduceNode(node0, &r1, &r2, &r3); in TEST_F()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/tir/
H A Dexpr_functor.h143 virtual R VisitExpr_(const ReduceNode* op, Args... args) EXPR_FUNCTOR_DEFAULT;
188 IR_EXPR_FUNCTOR_DISPATCH(ReduceNode); in InitVTable()
240 void VisitExpr_(const ReduceNode* op) override;
287 PrimExpr VisitExpr_(const ReduceNode* op) override;
H A Dexpr.h1023 class ReduceNode : public PrimExprNode {
1051 bool SEqualReduce(const ReduceNode* other, SEqualReducer equal) const { in SEqualReduce()
1070 TVM_DECLARE_FINAL_OBJECT_INFO(ReduceNode, PrimExprNode);
1082 TVM_DEFINE_OBJECT_REF_METHODS(Reduce, PrimExpr, ReduceNode);
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/arith/
H A Dir_mutator_with_analyzer.h60 PrimExpr VisitExpr_(const tir::ReduceNode* op) override;
H A Dir_visitor_with_analyzer.h55 void VisitExpr_(const ReduceNode* op) { in VisitExpr_()
H A Dcanonical_simplify.cc432 PrimExpr VisitExpr_(const ReduceNode* op) final;
535 PrimExpr SimplifyReduceCombiner(const ReduceNode* op);
980 PrimExpr CanonicalSimplifier::Impl::SimplifyReduceCombiner(const ReduceNode* op) { in SimplifyReduceCombiner()
1050 PrimExpr CanonicalSimplifier::Impl::VisitExpr_(const ReduceNode* op) { in VisitExpr_()
1053 op = ret.as<ReduceNode>(); in VisitExpr_()
/dports/www/node10/node-v10.24.1/deps/v8/src/compiler/
H A Ddead-code-elimination.h54 Reduction ReduceNode(Node* node); in NON_EXPORTED_BASE()
H A Dgraph-reducer.cc48 void GraphReducer::ReduceNode(Node* node) { in ReduceNode() function in v8::internal::compiler::GraphReducer
78 void GraphReducer::ReduceGraph() { ReduceNode(graph()->end()); } in ReduceGraph()
H A Dgraph-reducer.h140 void ReduceNode(Node* const); in NON_EXPORTED_BASE()
/dports/lang/v8/v8-9.6.180.12/src/compiler/
H A Ddead-code-elimination.h56 Reduction ReduceNode(Node* node); in NON_EXPORTED_BASE()
H A Dgraph-reducer.cc67 void GraphReducer::ReduceNode(Node* node) { in ReduceNode() function in v8::internal::compiler::GraphReducer
97 void GraphReducer::ReduceGraph() { ReduceNode(graph()->end()); } in ReduceGraph()
/dports/www/qt5-webengine/qtwebengine-everywhere-src-5.15.2/src/3rdparty/chromium/v8/src/compiler/
H A Ddead-code-elimination.h54 Reduction ReduceNode(Node* node); in NON_EXPORTED_BASE()
H A Dgraph-reducer.cc51 void GraphReducer::ReduceNode(Node* node) { in ReduceNode() function in v8::internal::compiler::GraphReducer
81 void GraphReducer::ReduceGraph() { ReduceNode(graph()->end()); } in ReduceGraph()
/dports/www/chromium-legacy/chromium-88.0.4324.182/v8/src/compiler/
H A Ddead-code-elimination.h56 Reduction ReduceNode(Node* node); in NON_EXPORTED_BASE()
H A Dgraph-reducer.cc54 void GraphReducer::ReduceNode(Node* node) { in ReduceNode() function in v8::internal::compiler::GraphReducer
84 void GraphReducer::ReduceGraph() { ReduceNode(graph()->end()); } in ReduceGraph()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/ir/
H A Dexpr_functor.cc87 void ExprVisitor::VisitExpr_(const ReduceNode* op) { in VisitExpr_()
216 PrimExpr ExprMutator::VisitExpr_(const ReduceNode* op) { in VisitExpr_()

123