Searched refs:reg_oi (Results 26 – 38 of 38) sorted by relevance
12
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_x8s8s32x_conv_kernel.hpp | 78 const Xbyak::Reg64 reg_oi = rbx; member
|
| H A D | jit_avx512_core_x8s8s32x_conv_kernel.hpp | 87 const Xbyak::Reg64 reg_oi = rbx; member
|
| H A D | jit_uni_x8s8s32x_conv_kernel.cpp | 1046 mov(reg_oi, n_urw_per_ow_block); in generate()
1064 cmove(reg_oi, reg_scratch); in generate()
1121 dec(reg_oi); in generate()
1147 mov(reg_oi, n_oi_middle_block_loop); in generate()
1154 dec(reg_oi); in generate()
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 66 const reg64_t reg_oi = r14; member
638 reg64_t reg_oi = r12; member
|
| H A D | jit_avx512_core_bf16_dw_conv_kernel.hpp | 60 reg64_t reg_oi = rbx; member
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 543 xor_(reg_oi, reg_oi); in ow_loop()
559 inc(reg_oi); in ow_loop()
569 inc(reg_oi); in ow_loop()
570 cmp(reg_oi, n_oi); in ow_loop()
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 62 reg64_t reg_oi = rbx; member
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_core_x8s8s32x_conv_kernel.hpp | 86 const Xbyak::Reg64 reg_oi = rbx; member
|
| H A D | jit_uni_x8s8s32x_conv_kernel.cpp | 1029 mov(reg_oi, n_urw_per_ow_block); in generate()
1047 cmove(reg_oi, reg_scratch); in generate()
1104 dec(reg_oi); in generate()
1130 mov(reg_oi, n_oi_middle_block_loop); in generate()
1137 dec(reg_oi); in generate()
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 66 const reg64_t reg_oi = r14; member
634 reg64_t reg_oi = r12; member
|
| H A D | jit_avx512_core_bf16_dw_conv_kernel.hpp | 60 reg64_t reg_oi = rbx; member
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 570 xor_(reg_oi, reg_oi); in ow_loop()
586 inc(reg_oi); in ow_loop()
596 inc(reg_oi); in ow_loop()
597 cmp(reg_oi, n_oi); in ow_loop()
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 62 reg64_t reg_oi = rbx; member
|
12