/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_sse41_conv_kernel_f32.cpp | 47 static constexpr size_t helper_vmm_idx = 15; in jit_sse41_conv_fwd_kernel_f32() local 52 helper_vmm_idx, r14, r15, preserve_gpr, preserve_vmm, in jit_sse41_conv_fwd_kernel_f32()
|
H A D | jit_gemm_x8s8s32x_convolution_utils.cpp | 181 static constexpr size_t helper_vmm_idx = 31; in jit_pp_ker_t() local 188 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_pp_ker_t()
|
H A D | jit_sse41_1x1_conv_kernel_f32.cpp | 50 static constexpr size_t helper_vmm_idx = 15; in jit_sse41_1x1_conv_kernel_f32() local 55 helper_vmm_idx, r13, r14, preserve_gpr, preserve_vmm, in jit_sse41_1x1_conv_kernel_f32()
|
H A D | jit_avx2_1x1_conv_kernel_f32.cpp | 57 static constexpr size_t helper_vmm_idx = 15; in jit_avx2_1x1_conv_kernel_f32() local 61 rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, r13, r14, in jit_avx2_1x1_conv_kernel_f32()
|
H A D | jit_gemm_inner_product_utils.cpp | 307 static const size_t helper_vmm_idx = is_avx512_ ? 31 : 15; in jit_pp_kernel_t() local 325 helper_vmm_idx, eltwise_reserved_gpr_, r14, preserve_gpr, in jit_pp_kernel_t()
|
H A D | jit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp | 56 static constexpr unsigned helper_vmm_idx = 31; in _jit_avx512_core_x8s8s32x_1x1_conv_kernel() local 63 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_avx512_core_x8s8s32x_1x1_conv_kernel()
|
H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 43 static constexpr size_t helper_vmm_idx = 31; in jit_avx512_dw_conv_fwd_kernel_bf16() local 48 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_avx512_dw_conv_fwd_kernel_bf16()
|
H A D | jit_avx512_common_1x1_conv_kernel.cpp | 58 static constexpr size_t helper_vmm_idx = 31; in jit_avx512_common_1x1_conv_kernel() local 62 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_avx512_common_1x1_conv_kernel()
|
H A D | jit_uni_x8s8s32x_conv_kernel.cpp | 64 static constexpr size_t helper_vmm_idx = 15; in _jit_uni_x8s8s32x_fwd_kernel() local 70 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_uni_x8s8s32x_fwd_kernel()
|
H A D | gemm_bf16_convolution.cpp | 92 static constexpr size_t helper_vmm_idx = 31; in pp_ker_t() local 96 helper_vmm_idx, reserved_eltwise_gpr, r14, preserve_gpr, in pp_ker_t()
|
H A D | jit_avx512_core_x8s8s32x_conv_kernel.cpp | 64 static constexpr size_t helper_vmm_idx = 31; in _jit_avx512_core_x8s8s32x_fwd_kernel() local 71 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_avx512_core_x8s8s32x_fwd_kernel()
|
H A D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 54 static constexpr size_t helper_vmm_idx = 31; in jit_avx512_core_bf16_1x1_conv_kernel() local 58 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_avx512_core_bf16_1x1_conv_kernel()
|
H A D | jit_uni_dw_conv_kernel_f32.cpp | 46 static constexpr size_t helper_vmm_idx = 31; in jit_uni_dw_conv_fwd_kernel_f32() local 50 rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, r14, r15, in jit_uni_dw_conv_fwd_kernel_f32()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_sse41_conv_kernel_f32.cpp | 47 static constexpr size_t helper_vmm_idx = 15; in jit_sse41_conv_fwd_kernel_f32() local 52 helper_vmm_idx, r14, r15, preserve_gpr, preserve_vmm, in jit_sse41_conv_fwd_kernel_f32()
|
H A D | jit_gemm_x8s8s32x_convolution_utils.cpp | 181 static constexpr size_t helper_vmm_idx = 31; in jit_pp_ker_t() local 188 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_pp_ker_t()
|
H A D | jit_sse41_1x1_conv_kernel_f32.cpp | 50 static constexpr size_t helper_vmm_idx = 15; in jit_sse41_1x1_conv_kernel_f32() local 55 helper_vmm_idx, r13, r14, preserve_gpr, preserve_vmm, in jit_sse41_1x1_conv_kernel_f32()
|
H A D | jit_avx2_1x1_conv_kernel_f32.cpp | 57 static constexpr size_t helper_vmm_idx = 15; in jit_avx2_1x1_conv_kernel_f32() local 61 rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, r13, r14, in jit_avx2_1x1_conv_kernel_f32()
|
H A D | jit_gemm_inner_product_utils.cpp | 309 static const size_t helper_vmm_idx = is_avx512_ ? 31 : 15; in jit_pp_kernel_t() local 327 helper_vmm_idx, eltwise_reserved_gpr_, r14, preserve_gpr, in jit_pp_kernel_t()
|
H A D | jit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp | 56 static constexpr unsigned helper_vmm_idx = 31; in _jit_avx512_core_x8s8s32x_1x1_conv_kernel() local 63 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_avx512_core_x8s8s32x_1x1_conv_kernel()
|
H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 43 static constexpr size_t helper_vmm_idx = 31; in jit_avx512_dw_conv_fwd_kernel_bf16() local 48 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_avx512_dw_conv_fwd_kernel_bf16()
|
H A D | gemm_bf16_convolution.cpp | 92 static constexpr size_t helper_vmm_idx = 31; in pp_ker_t() local 96 helper_vmm_idx, reserved_eltwise_gpr, r14, preserve_gpr, in pp_ker_t()
|
H A D | jit_avx512_common_1x1_conv_kernel.cpp | 58 static constexpr size_t helper_vmm_idx = 31; in jit_avx512_common_1x1_conv_kernel() local 62 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in jit_avx512_common_1x1_conv_kernel()
|
H A D | jit_uni_x8s8s32x_conv_kernel.cpp | 64 static constexpr size_t helper_vmm_idx = 15; in _jit_uni_x8s8s32x_fwd_kernel() local 70 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_uni_x8s8s32x_fwd_kernel()
|
H A D | jit_avx512_core_x8s8s32x_conv_kernel.cpp | 64 static constexpr size_t helper_vmm_idx = 31; in _jit_avx512_core_x8s8s32x_fwd_kernel() local 71 const rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, in _jit_avx512_core_x8s8s32x_fwd_kernel()
|
H A D | jit_uni_dw_conv_kernel_f32.cpp | 46 static constexpr size_t helper_vmm_idx = 31; in jit_uni_dw_conv_fwd_kernel_f32() local 50 rhs_arg_static_params_t rhs_arg_static_params {helper_vmm_idx, r14, r15, in jit_uni_dw_conv_fwd_kernel_f32()
|