Home
last modified time | relevance | path

Searched defs:xe_lp_x8s8x_convolution_fwd_t (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Dxe_lp_x8s8x_convolution.hpp34 struct xe_lp_x8s8x_convolution_fwd_t : public gpu_primitive_t { struct
36 struct pd_t : public gpu_convolution_fwd_pd_t {
112 status_t init(engine_t *engine) override { in init()
158 status_t execute(const exec_ctx_t &ctx) const override { in execute()
163 status_t init_res_storage( in init_res_storage()
188 const pd_t *pd() const { return (const pd_t *)gpu_primitive_t::pd().get(); } in pd()
189 compute::kernel_t kernel_;
190 compute::kernel_t src_compensation_kernel_;
191 enum { SCALES_ = 0 };
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Dxe_lp_x8s8x_convolution.hpp34 struct xe_lp_x8s8x_convolution_fwd_t : public gpu_primitive_t { struct
36 struct pd_t : public gpu_convolution_fwd_pd_t {
120 status_t init(engine_t *engine) override { in init()
166 status_t execute(const exec_ctx_t &ctx) const override { in execute()
171 status_t init_res_storage( in init_res_storage()
196 const pd_t *pd() const { return (const pd_t *)gpu_primitive_t::pd().get(); } in pd()
197 compute::kernel_t kernel_;
198 compute::kernel_t src_compensation_kernel_;
199 enum { SCALES_ = 0 };