Home
last modified time | relevance | path

Searched refs:oh_inner (Results 1 – 7 of 7) sorted by relevance

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/x86/
H A Dconv2d_avx_1x1.py92 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
101 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
108 s[CC].unroll(oh_inner)
115 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
117 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
144 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
146 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
159 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
167 s[CC].unroll(oh_inner)
171 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/x86/
H A Dconv2d_avx_1x1.py92 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
101 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
108 s[CC].unroll(oh_inner)
115 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
117 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
144 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
146 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
159 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
167 s[CC].unroll(oh_inner)
171 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/
H A Dconv2d_avx_1x1.py96 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
98 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
111 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
114 s[CC].reorder(oc_chunk, oh_outer, ow_outer, ic_chunk, ic_block, oh_inner, ow_inner, oc_block)
119 s[CC].unroll(oh_inner)
125 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
127 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
136 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
138 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/generic/
H A Dconv2d.py198 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
200 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
216 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
219 s[CC].reorder(oc_chunk, oh_outer, ow_outer, kh, kw, ic_outer, ic_f_inner, oh_inner,
226 s[CC].unroll(oh_inner)
230 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
232 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/generic/
H A Dconv2d.py198 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
200 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
216 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
219 s[CC].reorder(oc_chunk, oh_outer, ow_outer, kh, kw, ic_outer, ic_f_inner, oh_inner,
226 s[CC].unroll(oh_inner)
230 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
232 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/generic/
H A Dconv2d.py294 oh_outer, oh_inner = s[C].split(oh, factor=oh_factor)
296 s[C].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
312 oh_outer, oh_inner = s[CC].split(oh, factor=oh_factor)
323 oh_inner,
334 s[CC].unroll(oh_inner)
340 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
342 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
351 oh_outer, oh_inner = s[O].split(oh, factor=oh_factor)
353 s[O].reorder(oc_chunk, oh_outer, ow_outer, oh_inner, ow_inner, oc_block)
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp6429 expr_t oh_tg_blk_idx, oh_inner; in init_bwd_w() local
6440 gemm_schedule.split(oh, cfg_.oh_tg_blk, oh_tg_blk_idx, oh_inner); in init_bwd_w()
6456 gemm_schedule.reorder({od_inner, oh_inner, ow_inner, mb_thr_blk_idx}); in init_bwd_w()