/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_uni_dw_conv_kernel_utils.hpp | 42 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 D | jit_uni_softmax.cpp | 800 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 D | jit_uni_i8i8_pooling.hpp | 92 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
|
H A D | jit_uni_batch_normalization_s8.cpp | 574 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 D | lrn_avx512_nhwc_executor.hpp | 34 : 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 D | lrn_avx512_blocked_executor.hpp | 34 : 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 D | jit_uni_lrn.cpp | 46 , 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 D | jit_uni_lrn.hpp | 63 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 D | lrn_avx512_nhwc_executor.hpp | 34 : 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 D | lrn_avx512_blocked_executor.hpp | 34 : 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 D | jit_uni_lrn.cpp | 46 , 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 D | jit_uni_lrn.hpp | 63 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 D | jit_uni_dw_conv_kernel_utils.hpp | 43 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 D | jit_uni_softmax.cpp | 613 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 D | jit_uni_batch_normalization_s8.cpp | 379 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 D | jit_uni_i8i8_pooling.hpp | 91 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
|
H A D | jit_uni_i8i8_pooling.cpp | 832 : 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 D | jit_uni_dw_conv_kernel_utils.hpp | 43 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 D | jit_uni_softmax.cpp | 613 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 D | jit_uni_batch_normalization_s8.cpp | 379 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 D | jit_uni_i8i8_pooling.hpp | 91 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 D | jit_uni_dw_conv_kernel_utils.hpp | 48 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 D | jit_uni_softmax.cpp | 804 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 D | jit_uni_i8i8_pooling.hpp | 90 std::unique_ptr<jit_uni_i8i8_pooling_fwd_ker_t<isa>> ker_; member
|
H A D | jit_uni_batch_normalization_s8.cpp | 578 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
|