/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/prelu/ |
H A D | jit_uni_prelu_backward_kernel.hpp | 37 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t 38 const void *src = nullptr, *weights = nullptr, *dst_diff = nullptr; 39 void *src_diff = nullptr, *weights_diff = nullptr; 40 size_t compute_data_size = 0u;
|
H A D | jit_prelu_reduction_kernel.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_reduction_kernel_t 37 size_t reduction_blocks = 0; 38 const void *weights_diff_scratch = nullptr; 39 void *weights_diff = nullptr; 40 bool tail = false; 41 bool is_last_c_blk = false;
|
H A D | jit_uni_prelu_forward_kernel.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_forward_kernel_t 37 const void *src = nullptr, *weights = nullptr, *dst = nullptr; 38 size_t compute_data_size = 0u;
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/prelu/ |
H A D | jit_uni_prelu_backward_kernel.hpp | 37 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_backward_kernel_t 38 const void *src = nullptr, *weights = nullptr, *dst_diff = nullptr; 39 void *src_diff = nullptr, *weights_diff = nullptr; 40 size_t compute_data_size = 0u;
|
H A D | jit_prelu_reduction_kernel.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_reduction_kernel_t 37 size_t reduction_blocks = 0; 38 const void *weights_diff_scratch = nullptr; 39 void *weights_diff = nullptr; 40 bool tail = false; 41 bool is_last_c_blk = false;
|
H A D | jit_uni_prelu_forward_kernel.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_prelu_forward_kernel_t 37 const void *src = nullptr, *weights = nullptr, *dst = nullptr; 38 size_t compute_data_size = 0u;
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
H A D | jit_diff_weights_peephole.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_diff_weights_peephole_t 37 const void *c_states = nullptr; 38 const void *scratch_gates = nullptr; 39 void *dst = nullptr;
|
H A D | jit_brgemm_transpose.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_brgemm_transpose_t 37 const void *src = nullptr; 38 void *dst = nullptr;
|
H A D | jit_gates_reduction.hpp | 46 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_gates_reduction_t 47 const void *src = nullptr; 48 void *dst = nullptr;
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
H A D | jit_diff_weights_peephole.hpp | 36 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_diff_weights_peephole_t 37 const void *c_states = nullptr; 38 const void *scratch_gates = nullptr; 39 void *dst = nullptr;
|
H A D | jit_brgemm_transpose_single_row.hpp | 42 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_brgemm_transpose_single_row_t 43 const void *src = nullptr; 44 void *dst = nullptr;
|
H A D | jit_gates_reduction.hpp | 46 struct call_params_t { struct in dnnl::impl::cpu::x64::jit_gates_reduction_t 47 const void *src = nullptr; 48 void *dst = nullptr;
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_uni_tbb_batch_normalization.cpp | 372 size_t N, C, S; 727 size_t N, C, S; 728 const void *src, *dst; 730 const acc_data_t *mean, *var; 1013 size_t N, C, S; 1014 const void *src, *diff_src, *diff_dst; 1016 const acc_data_t *mean, *var; 1017 const acc_data_t *scale, *diff_scale, *diff_shift; 1299 size_t N, C, S; 1300 const void *src, *diff_dst; [all …]
|
H A D | jit_avx512_core_f32_wino_conv_2x3.cpp | 46 struct call_params_t { struct 47 const void *src; 48 const void *wino_src; 168 struct call_params_t { struct 170 const void *dst; 174 const void *bias; 175 const void *scales; 370 struct call_params_t { struct 371 const void *src; 372 const void *dst; [all …]
|
H A D | jit_uni_batch_normalization.cpp | 51 struct call_params_t { struct 53 size_t N_ithr, N_nthr; 54 size_t coff_max, soff_max; 55 size_t mb_stride_Bc, spat_size, spat_size_loc; 56 size_t S_s, S_tail; 58 acc_data_t chan_size, eps, one; 61 const acc_data_t *mean, *var; 64 const void *src, *dst; 65 const void *diff_src, *diff_dst; 66 const acc_data_t *rbuf1, *rbuf2; [all …]
|
H A D | jit_uni_batch_normalization_s8.cpp | 41 struct call_params_t { struct 43 size_t channel_offt_count, spat_offt_count; 44 float eps; 45 const float *scale, *shift, *mean, *var; 46 const data_t *src, *dst;
|
H A D | jit_avx512_core_u8s8s32x_wino_convolution.cpp | 62 struct call_params_t { struct 63 const void *src; 64 const void *wino_src; 255 struct call_params_t { struct 257 const void *dst; 261 const void *bias; 262 const void *scales; 511 struct call_params_t { struct 512 const void *src; 513 const void *dst; [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_uni_tbb_batch_normalization.cpp | 370 size_t N, C, S; 725 size_t N, C, S; 726 const void *src, *dst; 728 const acc_data_t *mean, *var; 1011 size_t N, C, S; 1012 const void *src, *diff_src, *diff_dst; 1014 const acc_data_t *mean, *var; 1015 const acc_data_t *scale, *diff_scale, *diff_shift; 1297 size_t N, C, S; 1298 const void *src, *diff_dst; [all …]
|
H A D | jit_avx512_core_f32_wino_conv_2x3.cpp | 46 struct call_params_t { struct 47 const void *src; 48 const void *wino_src; 168 struct call_params_t { struct 170 const void *dst; 174 const void *bias; 175 const void *scales; 370 struct call_params_t { struct 371 const void *src; 372 const void *dst; [all …]
|
H A D | jit_uni_batch_normalization.cpp | 50 struct call_params_t { struct 52 size_t N_ithr, N_nthr; 53 size_t coff_max, soff_max; 54 size_t mb_stride_Bc, spat_size, spat_size_loc; 55 size_t S_s, S_tail; 57 acc_data_t chan_size, eps, one; 60 const acc_data_t *mean, *var; 63 const void *src, *dst; 64 const void *diff_src, *diff_dst; 65 const acc_data_t *rbuf1, *rbuf2; [all …]
|
H A D | jit_uni_batch_normalization_s8.cpp | 39 struct call_params_t { struct 41 size_t channel_offt_count, spat_offt_count; 42 float eps; 43 const float *scale, *shift, *mean, *var; 44 const data_t *src, *dst;
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_uni_batch_normalization_s8.cpp | 44 struct call_params_t { struct 46 size_t channel_offt_count, spat_offt_count; 47 float eps; 48 const float *scale_shift, *mean, *var; 49 const data_t *src, *dst;
|
H A D | jit_uni_batch_normalization.cpp | 53 struct call_params_t { struct 55 size_t N_ithr, N_nthr; 56 size_t coff_max, soff_max; 57 size_t mb_stride_Bc, spat_size, spat_size_loc; 58 size_t S_s, S_tail; 60 acc_data_t chan_size, eps, one; 62 const acc_data_t *mean, *var; 64 const void *src, *dst; 65 const void *diff_src, *diff_dst; 66 const acc_data_t *rbuf1, *rbuf2; [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_uni_batch_normalization_s8.cpp | 44 struct call_params_t { struct 46 size_t channel_offt_count, spat_offt_count; 47 float eps; 48 const float *scale_shift, *mean, *var; 49 const data_t *src, *dst;
|
H A D | jit_uni_batch_normalization.cpp | 53 struct call_params_t { struct 55 size_t N_ithr, N_nthr; 56 size_t coff_max, soff_max; 57 size_t mb_stride_Bc, spat_size, spat_size_loc; 58 size_t S_s, S_tail; 60 acc_data_t chan_size, eps, one; 62 const acc_data_t *mean, *var; 64 const void *src, *dst; 65 const void *diff_src, *diff_dst; 66 const acc_data_t *rbuf1, *rbuf2; [all …]
|