| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 924 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1058 add_imm(reg_input_baddr, reg_input_baddr, in compute_ow_block_unroll()
1080 add_imm(reg_input_baddr, reg_input_baddr, in compute_ow_block_unroll()
1107 ldr(reg_input_baddr, in generate()
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 229 reg64_t reg_input_baddr = x10; member
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 924 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1058 add_imm(reg_input_baddr, reg_input_baddr, in compute_ow_block_unroll()
1080 add_imm(reg_input_baddr, reg_input_baddr, in compute_ow_block_unroll()
1107 ldr(reg_input_baddr, in generate()
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 229 reg64_t reg_input_baddr = x10; member
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 1306 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1462 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1478 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1498 mov(reg_input_baddr, ptr[this->param1 + GET_OFF(input)]); in generate()
|
| H A D | jit_avx512_core_bf16_dw_conv_kernel.hpp | 259 reg64_t reg_input_baddr = r15; member
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 256 reg64_t reg_input_baddr = r15; member
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 1822 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1981 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1997 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
2018 mov(reg_input_baddr, ptr[this->param1 + GET_OFF(input)]); in generate()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 1301 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1457 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1473 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1493 mov(reg_input_baddr, ptr[this->param1 + GET_OFF(input)]); in generate()
|
| H A D | jit_avx512_core_bf16_dw_conv_kernel.hpp | 262 reg64_t reg_input_baddr = r15; member
|
| H A D | jit_uni_dw_conv_kernel_f32.hpp | 257 reg64_t reg_input_baddr = r15; member
|
| H A D | jit_uni_dw_conv_kernel_f32.cpp | 1795 mov(reg_tmp_input, reg_input_baddr); in compute_h_loop()
1954 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1970 add(reg_input_baddr, data_offset * jcp.stride_w); in compute_ow_block_unroll()
1991 mov(reg_input_baddr, ptr[this->param1 + GET_OFF(input)]); in generate()
|