/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/ |
H A D | build_module.h | 197 int auto_unroll_max_step = 0; variable 239 v->Visit("auto_unroll_max_step", &auto_unroll_max_step); in VisitAttrs()
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/ |
H A D | build_module.h | 197 int auto_unroll_max_step = 0; variable 239 v->Visit("auto_unroll_max_step", &auto_unroll_max_step); in VisitAttrs()
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/reduce/ |
H A D | test_reduce_map.py | 66 with tvm.build_config(auto_unroll_max_step=16,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/reduce/ |
H A D | test_reduce_map.py | 66 with tvm.build_config(auto_unroll_max_step=16,
|
/dports/misc/tvm/incubator-tvm-0.6.1/tests/webgl/ |
H A D | test_local_topi_conv2d_nchw.py | 62 with tvm.build_config(auto_unroll_max_step=1400,
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/conv/ |
H A D | test_conv2d_hwcn_map.py | 79 with tvm.build_config(auto_unroll_max_step=128,
|
H A D | depthwise_conv2d_test.py | 131 with tvm.build_config(auto_unroll_max_step=128, 220 with tvm.build_config(auto_unroll_max_step=128,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/webgl/ |
H A D | test_local_topi_conv2d_nchw.py | 62 with tvm.build_config(auto_unroll_max_step=1400,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/conv/ |
H A D | test_conv2d_hwcn_map.py | 79 with tvm.build_config(auto_unroll_max_step=128,
|
H A D | depthwise_conv2d_test.py | 131 with tvm.build_config(auto_unroll_max_step=128, 220 with tvm.build_config(auto_unroll_max_step=128,
|
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/compiler/ |
H A D | build_module.py | 422 with tvm.build_config(auto_unroll_max_step=0): 460 with tvm.build_config(auto_unroll_max_step=0):
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/compiler/ |
H A D | build_module.py | 422 with tvm.build_config(auto_unroll_max_step=0): 460 with tvm.build_config(auto_unroll_max_step=0):
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/ |
H A D | loop_state.cc | 80 node->attrs.auto_unroll_max_step = 0; in Stage() 358 if (stage->attrs.auto_unroll_max_step != 0) { in PrintStage() 362 *os << stage->op->name << " auto_unroll: " << stage->attrs.auto_unroll_max_step << "\n"; in PrintStage()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/auto_scheduler/ |
H A D | loop_state.h | 88 int auto_unroll_max_step; member
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/rnn/ |
H A D | matexp.py | 131 auto_unroll_max_step=128,
|
H A D | lstm.py | 192 auto_unroll_max_step=128,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/rnn/ |
H A D | matexp.py | 131 auto_unroll_max_step=128,
|
H A D | lstm.py | 192 auto_unroll_max_step=128,
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/gemm/ |
H A D | cuda_gemm_square.py | 148 with tvm.build_config(auto_unroll_max_step=128,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/gemm/ |
H A D | cuda_gemm_square.py | 148 with tvm.build_config(auto_unroll_max_step=128,
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/codegen/ |
H A D | build_module.cc | 421 stmt = ir::UnrollLoop(stmt, config->auto_unroll_max_step, config->auto_unroll_max_depth, in BuildStmt() 664 p->stream << "auto_unroll_max_step=" << op->auto_unroll_max_step << ", "; in __anon58466f1f0402()
|
/dports/misc/tvm/incubator-tvm-0.6.1/src/codegen/ |
H A D | build_module.cc | 421 stmt = ir::UnrollLoop(stmt, config->auto_unroll_max_step, config->auto_unroll_max_depth, in BuildStmt() 664 p->stream << "auto_unroll_max_step=" << op->auto_unroll_max_step << ", "; in __anon299e7cc90402()
|
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/ |
H A D | build_module.py | 409 cfg.auto_unroll_max_step,
|
/dports/misc/tvm/incubator-tvm-0.6.1/tutorials/optimize/ |
H A D | opt_conv_tensorcore.py | 335 with tvm.build_config(auto_unroll_max_step=16):
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/ |
H A D | build_module.py | 409 cfg.auto_unroll_max_step,
|