Home
last modified time | relevance | path

Searched refs:VisitStmt_ (Results 1 – 25 of 119) sorted by relevance

12345

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/tir/
H A Dstmt_functor.h147 void VisitStmt_(const AttrStmtNode* op) override;
149 void VisitStmt_(const LetStmtNode* op) override;
150 void VisitStmt_(const ForNode* op) override;
151 void VisitStmt_(const AllocateNode* op) override;
152 void VisitStmt_(const StoreNode* op) override;
158 void VisitStmt_(const PrefetchNode* op) override;
159 void VisitStmt_(const SeqStmtNode* op) override;
240 Stmt VisitStmt_(const LetStmtNode* op) override;
241 Stmt VisitStmt_(const ForNode* op) override;
243 Stmt VisitStmt_(const StoreNode* op) override;
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/transforms/
H A Dremove_no_op.cc39 Stmt VisitStmt_(const LetStmtNode* op) final { in VisitStmt_() function in tvm::tir::NoOpRemover
40 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
44 Stmt VisitStmt_(const AttrStmtNode* op) final { in VisitStmt_() function in tvm::tir::NoOpRemover
48 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
53 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
73 Stmt VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::NoOpRemover
74 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
81 Stmt VisitStmt_(const AllocateNode* op) final { in VisitStmt_() function in tvm::tir::NoOpRemover
82 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
88 Stmt stmt = StmtMutator::VisitStmt_(op); in VisitStmt_()
[all …]
H A Dhoist_if_then_else.cc144 void VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::HoistCandidateSelector
153 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
163 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
176 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
180 return StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
184 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
190 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
223 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
366 Stmt VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::IfThenElseHoister
379 return StmtMutator::VisitStmt_(op); in VisitStmt_()
[all …]
H A Dir_util.cc121 Stmt VisitStmt_(const StoreNode* op) final { in VisitStmt_() function in tvm::tir::IRConvertSSA
122 Stmt stmt = StmtExprMutator::VisitStmt_(op); in VisitStmt_()
130 Stmt VisitStmt_(const LetStmtNode* op) final { in VisitStmt_() function in tvm::tir::IRConvertSSA
141 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
144 Stmt VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::IRConvertSSA
149 Stmt stmt = StmtExprMutator::VisitStmt_(op); in VisitStmt_()
155 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
158 Stmt VisitStmt_(const AllocateNode* op) final { in VisitStmt_() function in tvm::tir::IRConvertSSA
169 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
172 Stmt VisitStmt_(const AttrStmtNode* op) final { in VisitStmt_() function in tvm::tir::IRConvertSSA
[all …]
H A Dstorage_access.cc54 void StorageAccessVisitor::VisitStmt_(const StoreNode* op) { in VisitStmt_() function in tvm::tir::StorageAccessVisitor
71 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
83 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
96 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
101 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
118 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
126 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
132 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
136 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
140 void StorageAccessVisitor::VisitStmt_(const ForNode* op) { in VisitStmt_() function in tvm::tir::StorageAccessVisitor
[all …]
H A Dlower_warp_memory.cc114 void VisitStmt_(const StoreNode* op) final { in VisitStmt_() function in tvm::tir::WarpStoreCoeffFinder
126 StmtVisitor::VisitStmt_(op); in VisitStmt_()
194 StmtVisitor::VisitStmt_(op); in VisitStmt_()
316 void VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::BindVarBoundInfo
319 StmtVisitor::VisitStmt_(op); in VisitStmt_()
322 void VisitStmt_(const AttrStmtNode* op) { in VisitStmt_() function in tvm::tir::BindVarBoundInfo
332 StmtVisitor::VisitStmt_(op); in VisitStmt_()
356 Stmt VisitStmt_(const AllocateNode* op) { in VisitStmt_() function in tvm::tir::WarpMemoryRewriter
357 auto ret = StmtMutator::VisitStmt_(op); in VisitStmt_()
366 Stmt VisitStmt_(const AttrStmtNode* op) { in VisitStmt_() function in tvm::tir::WarpMemoryRewriter
[all …]
H A Dstorage_rewrite.cc93 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
95 void VisitStmt_(const StoreNode* op) final { in VisitStmt_() function in tvm::tir::LinearAccessPatternFinder
98 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
116 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
158 StmtExprVisitor::VisitStmt_(op); in VisitNewScope()
183 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
185 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
252 using StmtExprVisitor::VisitStmt_;
292 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
437 Stmt VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::StoragePlanRewriter
[all …]
H A Dinject_double_buffer.cc54 void VisitStmt_(const AttrStmtNode* op) final { in VisitStmt_() function in tvm::tir::DoubleBufferDetector
57 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
59 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
74 Stmt VisitStmt_(const AttrStmtNode* op) final { in VisitStmt_() function in tvm::tir::StripDoubleBufferWrite
78 return StmtMutator::VisitStmt_(op); in VisitStmt_()
105 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
110 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
134 return StmtExprMutator::VisitStmt_(op); in VisitStmt_()
138 Stmt VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::tir::DoubleBufferInjector
140 Stmt stmt = StmtExprMutator::VisitStmt_(op); in VisitStmt_()
[all …]
H A Dloop_partition.cc111 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
117 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
129 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
137 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
204 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
218 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
222 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
323 return StmtMutator::VisitStmt_(op); in VisitStmt_()
362 return StmtMutator::VisitStmt_(op); in VisitStmt_()
380 res = StmtMutator::VisitStmt_(op); in VisitStmt_()
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/codegen/stackvm/
H A Dcodegen_stackvm.h143 void VisitStmt_(const LetStmt* op) final;
144 void VisitStmt_(const Store* op) final;
145 void VisitStmt_(const For* op) final;
146 void VisitStmt_(const IfThenElse* op) final;
147 void VisitStmt_(const Allocate* op) final;
148 void VisitStmt_(const AttrStmt* op) final;
149 void VisitStmt_(const AssertStmt* op) final;
150 void VisitStmt_(const Evaluate* op) final;
151 void VisitStmt_(const Block* op) final;
152 void VisitStmt_(const ProducerConsumer* op) final;
/dports/misc/tvm/incubator-tvm-0.6.1/src/codegen/stackvm/
H A Dcodegen_stackvm.h143 void VisitStmt_(const LetStmt* op) final;
144 void VisitStmt_(const Store* op) final;
145 void VisitStmt_(const For* op) final;
146 void VisitStmt_(const IfThenElse* op) final;
147 void VisitStmt_(const Allocate* op) final;
148 void VisitStmt_(const AttrStmt* op) final;
149 void VisitStmt_(const AssertStmt* op) final;
150 void VisitStmt_(const Evaluate* op) final;
151 void VisitStmt_(const Block* op) final;
152 void VisitStmt_(const ProducerConsumer* op) final;
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/arith/
H A Ddomain_touched.cc53 void VisitStmt_(const ForNode* op) final { in VisitStmt_() function in tvm::arith::BufferTouchedDomain
56 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
60 void VisitStmt_(const LetStmtNode* op) final { in VisitStmt_() function in tvm::arith::BufferTouchedDomain
62 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
67 void VisitStmt_(const AttrStmtNode* op) final { in VisitStmt_() function in tvm::arith::BufferTouchedDomain
73 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
76 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
87 void VisitStmt_(const BufferStoreNode* op) final { in VisitStmt_() function in tvm::arith::BufferTouchedDomain
91 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
H A Dir_mutator_with_analyzer.h49 using StmtExprMutator::VisitStmt_;
52 tir::Stmt VisitStmt_(const tir::ForNode* op) override;
53 tir::Stmt VisitStmt_(const tir::LetStmtNode* op) override;
54 tir::Stmt VisitStmt_(const tir::IfThenElseNode* op) override;
55 tir::Stmt VisitStmt_(const tir::AttrStmtNode* op) override;
56 tir::Stmt VisitStmt_(const tir::AssertStmtNode* op) override;
H A Dir_visitor_with_analyzer.h39 void VisitStmt_(const ForNode* op) { in VisitStmt_() function
41 return StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
44 void VisitStmt_(const AttrStmtNode* op) { in VisitStmt_() function
49 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
51 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/ir/
H A Dstmt_functor.cc32 void StmtVisitor::VisitStmt_(const LetStmtNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
37 void StmtVisitor::VisitStmt_(const AttrStmtNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
42 void StmtVisitor::VisitStmt_(const ForNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
48 void StmtVisitor::VisitStmt_(const AllocateNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
54 void StmtVisitor::VisitStmt_(const StoreNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
102 void StmtVisitor::VisitStmt_(const PrefetchNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
109 void StmtVisitor::VisitStmt_(const SeqStmtNode* op) { in VisitStmt_() function in tvm::tir::StmtVisitor
167 Stmt StmtMutator::VisitStmt_(const ForNode* op) { in VisitStmt_() function in tvm::tir::StmtMutator
217 Stmt StmtMutator::VisitStmt_(const StoreNode* op) { in VisitStmt_() function in tvm::tir::StmtMutator
493 Stmt VisitStmt_(const StoreNode* op) final { in VisitStmt_() function in tvm::tir::IRSubstitue
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/contrib/hybrid/
H A Dcodegen_hybrid.h124 void VisitStmt_(const LetStmt* op) override;
125 void VisitStmt_(const Store* op) override;
126 void VisitStmt_(const Provide* op) override;
127 void VisitStmt_(const For* op) override;
128 void VisitStmt_(const IfThenElse* op) override;
129 void VisitStmt_(const Allocate* op) override;
130 void VisitStmt_(const Realize* op) override;
131 void VisitStmt_(const AttrStmt* op) override;
132 void VisitStmt_(const AssertStmt* op) override;
133 void VisitStmt_(const Evaluate* op) override;
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/src/contrib/hybrid/
H A Dcodegen_hybrid.h124 void VisitStmt_(const LetStmt* op) override;
125 void VisitStmt_(const Store* op) override;
126 void VisitStmt_(const Provide* op) override;
127 void VisitStmt_(const For* op) override;
128 void VisitStmt_(const IfThenElse* op) override;
129 void VisitStmt_(const Allocate* op) override;
130 void VisitStmt_(const Realize* op) override;
131 void VisitStmt_(const AttrStmt* op) override;
132 void VisitStmt_(const AssertStmt* op) override;
133 void VisitStmt_(const Evaluate* op) override;
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/codegen/spirv/
H A Dcodegen_spirv.h94 void VisitStmt_(const Store* op) override;
95 void VisitStmt_(const For* op) override;
96 void VisitStmt_(const IfThenElse* op) override;
97 void VisitStmt_(const Allocate* op) override;
98 void VisitStmt_(const AttrStmt* op) override;
99 void VisitStmt_(const AssertStmt* op) override;
100 void VisitStmt_(const LetStmt* op) override;
101 void VisitStmt_(const Block* op) override;
102 void VisitStmt_(const Evaluate* op) override;
103 void VisitStmt_(const ProducerConsumer* op) override;
/dports/misc/tvm/incubator-tvm-0.6.1/src/codegen/spirv/
H A Dcodegen_spirv.h94 void VisitStmt_(const Store* op) override;
95 void VisitStmt_(const For* op) override;
96 void VisitStmt_(const IfThenElse* op) override;
97 void VisitStmt_(const Allocate* op) override;
98 void VisitStmt_(const AttrStmt* op) override;
99 void VisitStmt_(const AssertStmt* op) override;
100 void VisitStmt_(const LetStmt* op) override;
101 void VisitStmt_(const Block* op) override;
102 void VisitStmt_(const Evaluate* op) override;
103 void VisitStmt_(const ProducerConsumer* op) override;
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/target/stackvm/
H A Dcodegen_stackvm.h137 void VisitStmt_(const LetStmtNode* op) final;
138 void VisitStmt_(const StoreNode* op) final;
139 void VisitStmt_(const ForNode* op) final;
140 void VisitStmt_(const IfThenElseNode* op) final;
141 void VisitStmt_(const AllocateNode* op) final;
142 void VisitStmt_(const AttrStmtNode* op) final;
143 void VisitStmt_(const AssertStmtNode* op) final;
144 void VisitStmt_(const EvaluateNode* op) final;
145 void VisitStmt_(const SeqStmtNode* op) final;
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/autotvm/
H A Dfeature_visitor.cc32 void FeatureVisitor::VisitStmt_(const ForNode* op) { in VisitStmt_() function in tvm::autotvm::FeatureVisitor
53 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
59 void FeatureVisitor::VisitStmt_(const AttrStmtNode* op) { in VisitStmt_() function in tvm::autotvm::FeatureVisitor
87 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
91 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
102 void FeatureVisitor::VisitStmt_(const StoreNode* op) { in VisitStmt_() function in tvm::autotvm::FeatureVisitor
104 StmtExprVisitor::VisitStmt_(op); in VisitStmt_()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/contrib/hybrid/
H A Dcodegen_hybrid.h121 void VisitStmt_(const LetStmtNode* op) override;
122 void VisitStmt_(const StoreNode* op) override;
123 void VisitStmt_(const ProducerStoreNode* op) override;
124 void VisitStmt_(const ForNode* op) override;
125 void VisitStmt_(const IfThenElseNode* op) override;
126 void VisitStmt_(const AllocateNode* op) override;
127 void VisitStmt_(const ProducerRealizeNode* op) override;
128 void VisitStmt_(const AttrStmtNode* op) override;
129 void VisitStmt_(const AssertStmtNode* op) override;
130 void VisitStmt_(const EvaluateNode* op) override;
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/
H A Dir_functor_ext.h100 return self->VisitStmt_(static_cast<const OP*>(n.get()), \
244 virtual R VisitStmt_(const LetStmt* op, Args... args) STMT_FUNCTOR_DEFAULT;
245 virtual R VisitStmt_(const AttrStmt* op, Args... args) STMT_FUNCTOR_DEFAULT;
247 virtual R VisitStmt_(const For* op, Args... args) STMT_FUNCTOR_DEFAULT;
248 virtual R VisitStmt_(const Allocate* op, Args... args) STMT_FUNCTOR_DEFAULT;
249 virtual R VisitStmt_(const Store* op, Args... args) STMT_FUNCTOR_DEFAULT;
250 virtual R VisitStmt_(const Free* op, Args... args) STMT_FUNCTOR_DEFAULT;
253 virtual R VisitStmt_(const Provide* op, Args... args) STMT_FUNCTOR_DEFAULT;
254 virtual R VisitStmt_(const Realize* op, Args... args) STMT_FUNCTOR_DEFAULT;
255 virtual R VisitStmt_(const Prefetch* op, Args... args) STMT_FUNCTOR_DEFAULT;
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/
H A Dir_functor_ext.h100 return self->VisitStmt_(static_cast<const OP*>(n.get()), \
244 virtual R VisitStmt_(const LetStmt* op, Args... args) STMT_FUNCTOR_DEFAULT;
245 virtual R VisitStmt_(const AttrStmt* op, Args... args) STMT_FUNCTOR_DEFAULT;
247 virtual R VisitStmt_(const For* op, Args... args) STMT_FUNCTOR_DEFAULT;
248 virtual R VisitStmt_(const Allocate* op, Args... args) STMT_FUNCTOR_DEFAULT;
249 virtual R VisitStmt_(const Store* op, Args... args) STMT_FUNCTOR_DEFAULT;
250 virtual R VisitStmt_(const Free* op, Args... args) STMT_FUNCTOR_DEFAULT;
253 virtual R VisitStmt_(const Provide* op, Args... args) STMT_FUNCTOR_DEFAULT;
254 virtual R VisitStmt_(const Realize* op, Args... args) STMT_FUNCTOR_DEFAULT;
255 virtual R VisitStmt_(const Prefetch* op, Args... args) STMT_FUNCTOR_DEFAULT;
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/target/spirv/
H A Dcodegen_spirv.h94 void VisitStmt_(const StoreNode* op) override;
95 void VisitStmt_(const ForNode* op) override;
96 void VisitStmt_(const IfThenElseNode* op) override;
97 void VisitStmt_(const AllocateNode* op) override;
98 void VisitStmt_(const AttrStmtNode* op) override;
99 void VisitStmt_(const AssertStmtNode* op) override;
100 void VisitStmt_(const LetStmtNode* op) override;
101 void VisitStmt_(const SeqStmtNode* op) override;
102 void VisitStmt_(const EvaluateNode* op) override;

12345