Home
last modified time | relevance | path

Searched defs:call_params_t (Results 1 – 25 of 38) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/prelu/
H A Djit_uni_prelu_backward_kernel.hpp37 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 Djit_prelu_reduction_kernel.hpp36 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 Djit_uni_prelu_forward_kernel.hpp36 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 Djit_uni_prelu_backward_kernel.hpp37 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 Djit_prelu_reduction_kernel.hpp36 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 Djit_uni_prelu_forward_kernel.hpp36 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 Djit_diff_weights_peephole.hpp36 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 Djit_brgemm_transpose.hpp36 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 Djit_gates_reduction.hpp46 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 Djit_diff_weights_peephole.hpp36 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 Djit_brgemm_transpose_single_row.hpp42 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 Djit_gates_reduction.hpp46 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 Djit_uni_tbb_batch_normalization.cpp372 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 Djit_avx512_core_f32_wino_conv_2x3.cpp46 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 Djit_uni_batch_normalization.cpp51 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 Djit_uni_batch_normalization_s8.cpp41 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 Djit_avx512_core_u8s8s32x_wino_convolution.cpp62 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 Djit_uni_tbb_batch_normalization.cpp370 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 Djit_avx512_core_f32_wino_conv_2x3.cpp46 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 Djit_uni_batch_normalization.cpp50 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 Djit_uni_batch_normalization_s8.cpp39 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 Djit_uni_batch_normalization_s8.cpp44 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 Djit_uni_batch_normalization.cpp53 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 Djit_uni_batch_normalization_s8.cpp44 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 Djit_uni_batch_normalization.cpp53 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 …]

12