/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/ |
H A D | conv2d_nhwc_winograd.py | 136 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 D | conv2d_nhwc.py | 115 s[AA].storage_align(xi, A_align - 1, A_align) 124 s[WW].storage_align(ic, W_align - 1, W_align)
|
H A D | conv2d_nhwc_tensorcore.py | 214 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 D | conv3d_ndhwc_tensorcore.py | 223 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 D | dense_tensorcore.py | 196 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 D | test_te_schedule_bound_inference_tiling.py | 42 s[C_shared].storage_align(ni, factor * 2, padding)
|
H A D | test_tir_transform_storage_flatten.py | 70 s[A1].storage_align(A1.op.axis[0], 2, 1)
|
H A D | test_auto_scheduler_measure.py | 165 s.storage_align(C, s[C].iters[-1], 8, 4)
|
/dports/misc/tvm/incubator-tvm-0.6.1/tutorials/optimize/ |
H A D | opt_matmul_auto_tensorcore.py | 131 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 D | opt_matmul_auto_tensorcore.py | 131 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 D | test_pass_storage_flatten.py | 60 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 D | loop_state.h | 387 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 D | schedule.py | 485 def storage_align(self, axis, factor, offset): member in Stage
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/unittest/ |
H A D | test_pass_storage_flatten.py | 60 s[A1].storage_align(A1.op.axis[0], 2, 1)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tutorials/optimize/ |
H A D | opt_matmul_auto_tensorcore.py | 150 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 D | schedule.py | 637 def storage_align(self, axis, factor, offset): member in Stage
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/ |
H A D | schedule.py | 637 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 D | loop_state.py | 404 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 D | loop_state.cc | 306 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 D | gemm_int8.py | 94 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 D | gemm_int8.py | 106 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 D | gemm_int8.py | 94 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 D | conv2d.py | 420 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 D | schedule.h | 247 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 D | schedule.h | 247 TVM_DLL Stage& storage_align(IterVar axis, int factor, int offset); //NOLINT(*)
|