Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / src / common / softmax_pd.hpp
index cb156e7..44032f7 100644 (file)
@@ -102,9 +102,9 @@ struct softmax_bwd_pd_t: public primitive_desc_t {
     virtual const memory_pd_t *output_pd(int index = 0) const override
     { return index == 0 ? diff_src_pd() : nullptr; }
 
-    virtual int n_inputs() const override { return 2; }
-    virtual int n_outputs() const override
-    { return 1 + (workspace_pd() != nullptr); }
+    virtual int n_inputs() const override
+    { return 2 + (workspace_pd() != nullptr); }
+    virtual int n_outputs() const override { return 1; }
 
     virtual status_t query(query_t what, int idx, void *result) const override
     {