Home
last modified time | relevance | path

Searched refs:storage_align (Results 1 – 25 of 34) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/
H A Dconv2d_nhwc_winograd.py136 s[CS].storage_align(bb, CS_align - 1, CS_align)
171 s[stage].storage_align(xo, strides - 1, strides)
278 s[AA].storage_align(k, AA_align - 1, AA_align)
283 s[BB].storage_align(m, BB_align - 1, BB_align)
481 s[indata_s].storage_align(c_is, data_align - 1, data_align)
486 s[indata_s].storage_align(nu_is, kernel_align - 1, kernel_align)
599 s[inverse_s].storage_align(p_inv_s, inverse_in_align - 1, inverse_in_align)
H A Dconv2d_nhwc.py115 s[AA].storage_align(xi, A_align - 1, A_align)
124 s[WW].storage_align(ic, W_align - 1, W_align)
H A Dconv2d_nhwc_tensorcore.py214 s[OL].storage_align(wc, CS_align - 1, CS_align)
255 s[AS].storage_align(w, AS_align - 1, AS_align)
269 s[WS].storage_align(ic, WS_align - 1, WS_align)
H A Dconv3d_ndhwc_tensorcore.py223 s[OL].storage_align(wc, CS_align - 1, CS_align)
264 s[AS].storage_align(w, AS_align - 1, AS_align)
278 s[WS].storage_align(ic, WS_align - 1, WS_align)
H A Ddense_tensorcore.py196 s[CS].storage_align(bb, CS_align - 1, CS_align)
231 s[stage].storage_align(xo, strides - 1, strides)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/
H A Dtest_te_schedule_bound_inference_tiling.py42 s[C_shared].storage_align(ni, factor * 2, padding)
H A Dtest_tir_transform_storage_flatten.py70 s[A1].storage_align(A1.op.axis[0], 2, 1)
H A Dtest_auto_scheduler_measure.py165 s.storage_align(C, s[C].iters[-1], 8, 4)
/dports/misc/tvm/incubator-tvm-0.6.1/tutorials/optimize/
H A Dopt_matmul_auto_tensorcore.py131 s[AA].storage_align(AA.op.axis[0], factor, offset)
135 s[BB].storage_align(BB.op.axis[0], factor, offset)
/dports/misc/py-tvm/incubator-tvm-0.6.1/tutorials/optimize/
H A Dopt_matmul_auto_tensorcore.py131 s[AA].storage_align(AA.op.axis[0], factor, offset)
135 s[BB].storage_align(BB.op.axis[0], factor, offset)
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_pass_storage_flatten.py60 s[A1].storage_align(A1.op.axis[0], 2, 1)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/auto_scheduler/
H A Dloop_state.h387 TVM_DLL void storage_align(int stage_id, const Iterator& it, int factor, int offset);
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/te/
H A Dschedule.py485 def storage_align(self, axis, factor, offset): member in Stage
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/unittest/
H A Dtest_pass_storage_flatten.py60 s[A1].storage_align(A1.op.axis[0], 2, 1)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tutorials/optimize/
H A Dopt_matmul_auto_tensorcore.py150 s[AA].storage_align(AA.op.axis[0], factor, offset)
154 s[BB].storage_align(BB.op.axis[0], factor, offset)
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/
H A Dschedule.py637 def storage_align(self, axis, factor, offset): member in Stage
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/
H A Dschedule.py637 def storage_align(self, axis, factor, offset): member in Stage
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/auto_scheduler/
H A Dloop_state.py404 def storage_align(self, stage, iterator, factor, offset): member in State
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/
H A Dloop_state.cc306 void State::storage_align(int stage_id, const Iterator& it, int factor, int offset) { in storage_align() function in tvm::auto_scheduler::State
520 state.storage_align(stage_id, it, factor, offset); in __anon42f1274b0c02()
/dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/gemm/
H A Dgemm_int8.py94 s[stage].storage_align(s[stage].op.axis[0],
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/apps/topi_recipe/gemm/
H A Dgemm_int8.py106 s[stage].storage_align(s[stage].op.axis[0], cfg["storage_align"].val, 0)
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/gemm/
H A Dgemm_int8.py94 s[stage].storage_align(s[stage].op.axis[0],
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/intel_graphics/
H A Dconv2d.py420 s[temp_W].storage_align(s[temp_W].op.axis[2], 16, 0)
639 s[temp_W].storage_align(s[temp_W].op.axis[2], 16, 0)
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/
H A Dschedule.h247 TVM_DLL Stage& storage_align(IterVar axis, int factor, int offset); //NOLINT(*)
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/
H A Dschedule.h247 TVM_DLL Stage& storage_align(IterVar axis, int factor, int offset); //NOLINT(*)

12