Searched defs:jit_avx2_convolution_fwd_t (Results 1 – 2 of 2) sorted by relevance
36 struct jit_avx2_convolution_fwd_t : public primitive_t { struct37 struct pd_t : public cpu_convolution_fwd_pd_t {87 jit_avx2_convolution_fwd_t(const pd_t *apd) : primitive_t(apd) {} in jit_avx2_convolution_fwd_t() argument89 typedef typename prec_traits<data_type::f32>::type data_t;91 status_t init(engine_t *engine) override { in init()98 status_t execute(const exec_ctx_t &ctx) const override { in execute()105 const pd_t *pd() const { return (const pd_t *)primitive_t::pd().get(); } in pd()107 std::unique_ptr<jit_avx2_conv_fwd_kernel_f32> kernel_;