/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/tir/ |
H A D | stmt_functor.h | 147 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 D | remove_no_op.cc | 39 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 D | hoist_if_then_else.cc | 144 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 D | ir_util.cc | 121 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 D | storage_access.cc | 54 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 D | lower_warp_memory.cc | 114 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 D | storage_rewrite.cc | 93 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 D | inject_double_buffer.cc | 54 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 D | loop_partition.cc | 111 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 D | codegen_stackvm.h | 143 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 D | codegen_stackvm.h | 143 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 D | domain_touched.cc | 53 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 D | ir_mutator_with_analyzer.h | 49 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 D | ir_visitor_with_analyzer.h | 39 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 D | stmt_functor.cc | 32 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 D | codegen_hybrid.h | 124 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 D | codegen_hybrid.h | 124 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 D | codegen_spirv.h | 94 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 D | codegen_spirv.h | 94 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 D | codegen_stackvm.h | 137 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 D | feature_visitor.cc | 32 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 D | codegen_hybrid.h | 121 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 D | ir_functor_ext.h | 100 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 D | ir_functor_ext.h | 100 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 D | codegen_spirv.h | 94 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;
|