| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/ |
| H A D | transform_step.cc | 569 node->pragma_type = std::move(pragma_type); in PragmaStep()
586 node->pragma_type = std::move(string_value); in PragmaStep()
596 writer->WriteString(pragma_type); in WriteToRecord()
600 if (pragma_type == "debug_skip_region") { in ApplyToState()
607 for (; pos < pragma_type.size(); ++pos) { in ApplyToState()
608 if ((*(pragma_type.c_str() + pos)) == '$') { in ApplyToState()
626 for (; pos < pragma_type.size(); ++pos) { in ApplyToSchedule()
627 if ((*(pragma_type.c_str() + pos)) == '$') { in ApplyToSchedule()
636 stage.pragma(axes[iter_id], pragma_type); in ApplyToSchedule()
649 for (; pos < pragma_type.size(); ++pos) { in PrintAsPythonAPI()
[all …]
|
| H A D | loop_state.cc | 259 void State::pragma(int stage_id, const Iterator& it, const String& pragma_type) { in pragma() argument
261 PragmaStep step = PragmaStep(stage_id, GetIndex(stage->iters, it), pragma_type); in pragma()
485 .set_body_typed([](State state, int stage_id, const Iterator& it, const String& pragma_type) { in __anon42f1274b0702() argument
486 state.pragma(stage_id, it, pragma_type); in __anon42f1274b0702()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/te/ |
| H A D | schedule.py | 419 def pragma(self, var, pragma_type, pragma_value=None): argument
469 _ffi_api.StagePragma(self, var, pragma_type, pragma_value)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | schedule.py | 571 def pragma(self, var, pragma_type, pragma_value=None): argument
621 _api_internal._StagePragma(self, var, pragma_type, pragma_value)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | schedule.py | 571 def pragma(self, var, pragma_type, pragma_value=None): argument
621 _api_internal._StagePragma(self, var, pragma_type, pragma_value)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/auto_scheduler/ |
| H A D | loop_state.py | 262 def pragma(self, stage, iterator, pragma_type): argument
276 self.state_object, self._resolve_stage_id(stage), iterator, pragma_type
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | schedule_lang.cc | 403 const std::string& pragma_type, in pragma() argument
405 if (pragma_type == "unroll") { in pragma()
407 } else if (pragma_type == "vectorize") { in pragma()
411 operator->(), var, [pragma_type, pragma_value](IterVarAttrNode* n) { in pragma()
412 n->pragma_keys.push_back(ir::StringImm::make(pragma_type)); in pragma()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | schedule_lang.cc | 403 const std::string& pragma_type, in pragma() argument
405 if (pragma_type == "unroll") { in pragma()
407 } else if (pragma_type == "vectorize") { in pragma()
411 operator->(), var, [pragma_type, pragma_value](IterVarAttrNode* n) { in pragma()
412 n->pragma_keys.push_back(ir::StringImm::make(pragma_type)); in pragma()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/ |
| H A D | schedule_lang.cc | 374 Stage& Stage::pragma(IterVar var, const std::string& pragma_type, in pragma() argument
376 if (pragma_type == "unroll") { in pragma()
378 } else if (pragma_type == "vectorize") { in pragma()
381 UpdateIterVarAttr(operator->(), var, [pragma_type, pragma_value](IterVarAttrNode* n) { in pragma()
382 n->pragma_keys.push_back(tir::StringImm(pragma_type)); in pragma()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/auto_scheduler/ |
| H A D | loop_state.h | 336 TVM_DLL void pragma(int stage_id, const Iterator& it, const String& pragma_type);
|
| H A D | transform_step.h | 365 String pragma_type; variable
408 PragmaStep(int stage_id, int iter_id, String pragma_type);
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/search_policy/ |
| H A D | sketch_policy_rules.cc | 969 if (StrStartsWith(ps->pragma_type, "auto_unroll_max_step")) { in Apply()
1147 step = PragmaStep(ps->stage_id, 0, ps->pragma_type); in Apply()
1150 step = PragmaStep(ps->stage_id, ps->iter_id + iter_offset, ps->pragma_type); in Apply()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | schedule.h | 227 const std::string& pragma_type,
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | schedule.h | 227 const std::string& pragma_type,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/te/ |
| H A D | schedule.h | 228 TVM_DLL Stage& pragma(IterVar var, const std::string& pragma_type,
|
| /dports/lang/yap/yap-6.2.2/packages/chr/ |
| H A D | chr_translate.chr | 186 :- chr_type pragma_rule ---> pragma(rule,idspair,list(pragma_type),maybe(rule_name),rule_nb).
190 :- chr_type pragma_type ---> passive(id)
|
| /dports/lang/swi-pl/swipl-8.2.3/packages/chr/ |
| H A D | chr_translate.chr | 190 :- chr_type pragma_rule ---> pragma(rule,idspair,list(pragma_type),maybe(rule_name),rule_nb).
194 :- chr_type pragma_type ---> passive(id)
|