Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / src / cpu / jit_avx512_common_lrn.hpp
index 10b5bb8..8ec624a 100644 (file)
@@ -39,20 +39,20 @@ struct jit_avx512_common_lrn_fwd_t: public cpu_primitive_t {
         virtual status_t init() override;
     };
 
-    jit_avx512_common_lrn_fwd_t(const pd_t *pd, const input_vector &inputs,
+    jit_avx512_common_lrn_fwd_t(const pd_t *apd, const input_vector &inputs,
             const output_vector &outputs);
     ~jit_avx512_common_lrn_fwd_t();
 
     typedef typename prec_traits<data_type::f32>::type data_t;
 
-    virtual void execute(event_t *e) {
+    virtual void execute(event_t *e) const {
         execute_forward();
         e->set_state(event_t::ready);
     }
 
 private:
-    void execute_forward();
-    pd_t conf_;
+    void execute_forward() const;
+    const pd_t *pd() const { return (const pd_t *)primitive_t::pd(); }
 
     int use_h_parallelism;
     struct jit_avx512_common_lrn_kernel_f32;
@@ -73,20 +73,20 @@ struct jit_avx512_common_lrn_bwd_t: public cpu_primitive_t {
         virtual status_t init() override;
     };
 
-    jit_avx512_common_lrn_bwd_t(const pd_t *pd, const input_vector &inputs,
+    jit_avx512_common_lrn_bwd_t(const pd_t *apd, const input_vector &inputs,
             const output_vector &outputs);
     ~jit_avx512_common_lrn_bwd_t();
 
     typedef typename prec_traits<data_type::f32>::type data_t;
 
-    virtual void execute(event_t *e) {
+    virtual void execute(event_t *e) const {
         execute_backward();
         e->set_state(event_t::ready);
     }
 
 private:
-    void execute_backward();
-    pd_t conf_;
+    void execute_backward() const;
+    const pd_t *pd() const { return (const pd_t *)primitive_t::pd(); }
 
     int use_h_parallelism;
     struct jit_avx512_common_lrn_kernel_f32;