Home
last modified time | relevance | path

Searched refs:ker_ (Results 1 – 25 of 39) sorted by relevance

12

/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_dw_conv_kernel_utils.hpp42 ker_ = new jit_kernel_t(ajcp, dst_md); in jit_uni_dw_conv_fwd_kernel()
46 ~jit_uni_dw_conv_fwd_kernel() { delete ker_; } in ~jit_uni_dw_conv_fwd_kernel()
56 jit_generator *ker() const { return ker_; } in ker()
65 jit_kernel_t *ker_; member
72 : ker_(nullptr) { in jit_uni_dw_conv_bwd_data_kernel()
73 ker_ = new jit_kernel_t(ajcp); in jit_uni_dw_conv_bwd_data_kernel()
77 ~jit_uni_dw_conv_bwd_data_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_data_kernel()
93 jit_kernel_t *ker_; member
102 : ker_(nullptr) { in jit_uni_dw_conv_bwd_weights_kernel()
103 ker_ = new jit_kernel_t(ajcp); in jit_uni_dw_conv_bwd_weights_kernel()
[all …]
H A Djit_uni_softmax.cpp800 driver_t(const softmax_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
804 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
807 ker_(&p); in exec()
813 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
817 ker_(&p); in exec()
820 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
824 jit_softmax_t<isa> ker_; member
H A Djit_uni_i8i8_pooling.hpp92 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
H A Djit_uni_batch_normalization_s8.cpp574 driver_t(const batch_normalization_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
606 if (p.spat_offt_count != 0) ker_(&p); in exec()
609 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
614 jit_bnorm_s8_t<isa> ker_; member
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/lrn/
H A Dlrn_avx512_nhwc_executor.hpp34 : ker_(utils::make_unique< in lrn_avx512_nhwc_executor_fwd_t()
47 status_t create_kernel() override { return ker_->create_kernel(); } in create_kernel()
57 const auto ker = ker_.get(); in execute()
79 std::unique_ptr<jit_avx512_common_lrn_kernel_fwd_nhwc_t<d_type>> ker_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_nhwc_executor_fwd_t
89 : ker_ {utils::make_unique< in lrn_avx512_nhwc_executor_bwd_t()
99 status_t create_kernel() override { return ker_->create_kernel(); } in create_kernel()
109 const auto ker = ker_.get(); in execute()
132 std::unique_ptr<jit_avx512_common_lrn_kernel_bwd_nhwc_t<d_type>> ker_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_nhwc_executor_bwd_t
H A Dlrn_avx512_blocked_executor.hpp34 : ker_(nullptr) in lrn_avx512_blocked_executor_fwd_t()
50 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_fwd_t()
55 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_fwd_t()
73 CHECK(ker_->create_kernel()); in create_kernel()
87 const auto ker = ker_.get(); in execute()
161 ker_, ker_first_, ker_last_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_blocked_executor_fwd_t
174 : ker_(nullptr) in lrn_avx512_blocked_executor_bwd_t()
188 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_bwd_t()
193 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_bwd_t()
211 CHECK(ker_->create_kernel()); in create_kernel()
[all …]
H A Djit_uni_lrn.cpp46 , ker_(nullptr) in jit_uni_lrn_fwd_t()
69 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
78 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
81 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
90 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
93 CHECK(ker_->create_kernel()); in init()
120 const auto ker = ker_.get(); in execute_forward()
222 , ker_(nullptr) in jit_uni_lrn_bwd_t()
243 ker_ = utils::make_unique<jit_uni_lrn_bwd_kernel_t<isa, d_type>>( in init()
261 CHECK(ker_->create_kernel()); in init()
[all …]
H A Djit_uni_lrn.hpp63 std::unique_ptr<jit_uni_lrn_fwd_kernel_t<isa, d_type>> ker_, ker_first_, member
97 std::unique_ptr<jit_uni_lrn_bwd_kernel_t<isa, d_type>> ker_, ker_first_, member
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/lrn/
H A Dlrn_avx512_nhwc_executor.hpp34 : ker_(utils::make_unique< in lrn_avx512_nhwc_executor_fwd_t()
47 status_t create_kernel() override { return ker_->create_kernel(); } in create_kernel()
57 const auto ker = ker_.get(); in execute()
79 std::unique_ptr<jit_avx512_common_lrn_kernel_fwd_nhwc_t<d_type>> ker_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_nhwc_executor_fwd_t
89 : ker_ {utils::make_unique< in lrn_avx512_nhwc_executor_bwd_t()
99 status_t create_kernel() override { return ker_->create_kernel(); } in create_kernel()
109 const auto ker = ker_.get(); in execute()
132 std::unique_ptr<jit_avx512_common_lrn_kernel_bwd_nhwc_t<d_type>> ker_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_nhwc_executor_bwd_t
H A Dlrn_avx512_blocked_executor.hpp34 : ker_(nullptr) in lrn_avx512_blocked_executor_fwd_t()
50 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_fwd_t()
55 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_fwd_t()
73 CHECK(ker_->create_kernel()); in create_kernel()
87 const auto ker = ker_.get(); in execute()
161 ker_, ker_first_, ker_last_; member in dnnl::impl::cpu::x64::lrn::lrn_avx512_blocked_executor_fwd_t
174 : ker_(nullptr) in lrn_avx512_blocked_executor_bwd_t()
188 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_bwd_t()
193 ker_ = utils::make_unique< in lrn_avx512_blocked_executor_bwd_t()
211 CHECK(ker_->create_kernel()); in create_kernel()
[all …]
H A Djit_uni_lrn.cpp46 , ker_(nullptr) in jit_uni_lrn_fwd_t()
69 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
78 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
81 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
90 ker_ = utils::make_unique<jit_uni_lrn_fwd_kernel_t<isa, d_type>>( in init()
93 CHECK(ker_->create_kernel()); in init()
120 const auto ker = ker_.get(); in execute_forward()
222 , ker_(nullptr) in jit_uni_lrn_bwd_t()
243 ker_ = utils::make_unique<jit_uni_lrn_bwd_kernel_t<isa, d_type>>( in init()
261 CHECK(ker_->create_kernel()); in init()
[all …]
H A Djit_uni_lrn.hpp63 std::unique_ptr<jit_uni_lrn_fwd_kernel_t<isa, d_type>> ker_, ker_first_, member
97 std::unique_ptr<jit_uni_lrn_bwd_kernel_t<isa, d_type>> ker_, ker_first_, member
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_uni_dw_conv_kernel_utils.hpp43 ker_ = new jit_uni_dw_conv_fwd_kernel_f32<isa>(ajcp); in jit_uni_dw_conv_fwd_kernel()
45 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
46 ~jit_uni_dw_conv_fwd_kernel() { delete ker_; } in ~jit_uni_dw_conv_fwd_kernel()
58 jit_generator *ker() const { return ker_; } in ker()
62 jit_uni_dw_conv_fwd_kernel_f32<isa> *ker_; member
270 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
271 ~jit_uni_dw_conv_bwd_data_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_data_kernel()
284 jit_uni_dw_conv_bwd_data_kernel_f32<isa> *ker_; member
399 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
400 ~jit_uni_dw_conv_bwd_weights_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_weights_kernel()
[all …]
H A Djit_uni_softmax.cpp613 driver_t(const softmax_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
617 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
620 ker_(&p); in exec()
626 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
630 ker_(&p); in exec()
633 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
637 jit_softmax_t<isa> ker_; member
H A Djit_uni_batch_normalization_s8.cpp379 driver_t(const batch_normalization_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
409 if (p.spat_offt_count != 0) ker_(&p); in exec()
412 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
417 jit_bnorm_t<isa> ker_; member
H A Djit_uni_i8i8_pooling.hpp91 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
H A Djit_uni_i8i8_pooling.cpp832 : primitive_t(apd), ker_(nullptr) {} in jit_uni_i8i8_pooling_fwd_t()
839 CHECK(safe_ptr_assign(ker_, in init()
842 return ker_->create_kernel(); in init()
900 (*ker_)(&p); in execute_forward()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_uni_dw_conv_kernel_utils.hpp43 ker_ = new jit_uni_dw_conv_fwd_kernel_f32<isa>(ajcp); in jit_uni_dw_conv_fwd_kernel()
45 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
46 ~jit_uni_dw_conv_fwd_kernel() { delete ker_; } in ~jit_uni_dw_conv_fwd_kernel()
58 jit_generator *ker() const { return ker_; } in ker()
62 jit_uni_dw_conv_fwd_kernel_f32<isa> *ker_; member
270 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
271 ~jit_uni_dw_conv_bwd_data_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_data_kernel()
284 jit_uni_dw_conv_bwd_data_kernel_f32<isa> *ker_; member
399 status_t create_kernel() { return ker_->create_kernel(); } in create_kernel()
400 ~jit_uni_dw_conv_bwd_weights_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_weights_kernel()
[all …]
H A Djit_uni_softmax.cpp613 driver_t(const softmax_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
617 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
620 ker_(&p); in exec()
626 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
630 ker_(&p); in exec()
633 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
637 jit_softmax_t<isa> ker_; member
H A Djit_uni_batch_normalization_s8.cpp379 driver_t(const batch_normalization_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
409 if (p.spat_offt_count != 0) ker_(&p); in exec()
412 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
417 jit_bnorm_t<isa> ker_; member
H A Djit_uni_i8i8_pooling.hpp91 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_dw_conv_kernel_utils.hpp48 ker_ = new jit_kernel_t(ajcp, dst_md); in jit_uni_dw_conv_fwd_kernel()
52 ~jit_uni_dw_conv_fwd_kernel() { delete ker_; } in ~jit_uni_dw_conv_fwd_kernel()
62 jit_generator *ker() const { return ker_; } in ker()
71 jit_kernel_t *ker_; member
294 : ker_(nullptr) { in jit_uni_dw_conv_bwd_data_kernel()
295 ker_ = new jit_kernel_t(ajcp); in jit_uni_dw_conv_bwd_data_kernel()
299 ~jit_uni_dw_conv_bwd_data_kernel() { delete ker_; } in ~jit_uni_dw_conv_bwd_data_kernel()
315 jit_kernel_t *ker_; member
475 : ker_(nullptr) { in jit_uni_dw_conv_bwd_weights_kernel()
476 ker_ = new jit_kernel_t(ajcp); in jit_uni_dw_conv_bwd_weights_kernel()
[all …]
H A Djit_uni_softmax.cpp804 driver_t(const softmax_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
808 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
811 ker_(&p); in exec()
817 p.spat_offt_count = outer_stride * ker_.data_type_size_; in exec()
821 ker_(&p); in exec()
824 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
828 jit_softmax_t<isa> ker_; member
H A Djit_uni_i8i8_pooling.hpp90 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
H A Djit_uni_batch_normalization_s8.cpp578 driver_t(const batch_normalization_pd_t *pd) : pd_(pd), ker_(pd_) {} in driver_t()
610 if (p.spat_offt_count != 0) ker_(&p); in exec()
613 status_t create_kernel() { return ker_.create_kernel(); } in create_kernel()
618 jit_bnorm_t<isa> ker_; member

12