Home
last modified time | relevance | path

Searched refs:pragma_keys (Results 1 – 12 of 12) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/operation/
H A Dop_util.cc103 CHECK_EQ(it_attr->pragma_keys.size(), it_attr->pragma_values.size()); in MakeLoopNest()
104 for (size_t k = 0; k < it_attr->pragma_keys.size(); ++k) { in MakeLoopNest()
105 const std::string& pkey = it_attr->pragma_keys[k].as<StringImmNode>()->value; in MakeLoopNest()
/dports/misc/tvm/incubator-tvm-0.6.1/src/op/
H A Dop_util.cc96 CHECK_EQ(it_attr->pragma_keys.size(), it_attr->pragma_values.size()); in MakeLoopNest()
97 for (size_t k = 0; k < it_attr->pragma_keys.size(); ++k) { in MakeLoopNest()
98 const std::string& pkey = it_attr->pragma_keys[k].as<StringImm>()->value; in MakeLoopNest()
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/op/
H A Dop_util.cc96 CHECK_EQ(it_attr->pragma_keys.size(), it_attr->pragma_values.size()); in MakeLoopNest()
97 for (size_t k = 0; k < it_attr->pragma_keys.size(); ++k) { in MakeLoopNest()
98 const std::string& pkey = it_attr->pragma_keys[k].as<StringImm>()->value; in MakeLoopNest()
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/
H A Dschedule.h614 Array<Expr> pragma_keys; variable
628 v->Visit("pragma_keys", &pragma_keys); in VisitAttrs()
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/
H A Dschedule.h614 Array<Expr> pragma_keys; variable
628 v->Visit("pragma_keys", &pragma_keys); in VisitAttrs()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/te/
H A Dschedule.h596 Array<PrimExpr> pragma_keys; variable
610 v->Visit("pragma_keys", &pragma_keys); in VisitAttrs()
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_lang_schedule.py154 assert s[T].iter_var_attrs[xo].pragma_keys[0].value == "pragma1"
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_lang_schedule.py154 assert s[T].iter_var_attrs[xo].pragma_keys[0].value == "pragma1"
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/
H A Dtest_te_schedule.py204 assert s[T].iter_var_attrs[xo].pragma_keys[0].value == "pragma1"
/dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/
H A Dschedule_lang.cc412 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 Dschedule_lang.cc412 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 Dschedule_lang.cc382 n->pragma_keys.push_back(tir::StringImm(pragma_type)); in pragma()