Searched refs:InjectScanStep (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/ |
| H A D | schedule_ops.cc | 102 class InjectScanStep : public StmtMutator { class
104 InjectScanStep(const Stage& stage, const Operation& scan_op, in InjectScanStep() function in tvm::te::InjectScanStep
351 InjectScanStep mu(s, scan_init.at(s->op), dom_map, true, debug_keep_trivial_loop); in ScheduleOps()
357 InjectScanStep mu(s, attach_spec->attach_stage->op, dom_map, false, debug_keep_trivial_loop); in ScheduleOps()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | schedule_ops.cc | 118 class InjectScanStep : public IRMutator { class
120 InjectScanStep(const Stage& stage, in InjectScanStep() function in tvm::schedule::InjectScanStep
394 InjectScanStep mu(s, scan_init.at(s->op), dom_map, true, debug_keep_trivial_loop); in ScheduleOps()
401 InjectScanStep mu(s, attach_spec->attach_stage->op, dom_map, false, debug_keep_trivial_loop); in ScheduleOps()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | schedule_ops.cc | 118 class InjectScanStep : public IRMutator { class
120 InjectScanStep(const Stage& stage, in InjectScanStep() function in tvm::schedule::InjectScanStep
394 InjectScanStep mu(s, scan_init.at(s->op), dom_map, true, debug_keep_trivial_loop); in ScheduleOps()
401 InjectScanStep mu(s, attach_spec->attach_stage->op, dom_map, false, debug_keep_trivial_loop); in ScheduleOps()
|