Publishing 2019 R1.1 content and Myriad plugin sources (#162)

[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / src / common / binarization_pd.hpp

diff --git a/inference-engine/thirdparty/mkl-dnn/src/common/binarization_pd.hpp b/inference-engine/thirdparty/mkl-dnn/src/common/binarization_pd.hpp
index 1450230..95016b9 100644 (file)
--- a/inference-engine/thirdparty/mkl-dnn/src/common/binarization_pd.hpp
+++ b/inference-engine/thirdparty/mkl-dnn/src/common/binarization_pd.hpp
@@ -47,14 +47,14 @@ struct binarization_fwd_pd_t: public primitive_desc_t {
     virtual const memory_pd_t *input_pd(int index = 0) const override {
         switch (index) {
         case 0: return src_pd();
-        case 1: return weights_pd(index - 1);
+        case 1: case 2: return weights_pd(index - 1);
         default: return nullptr;
         }
     }
     virtual const memory_pd_t *output_pd(int index = 0) const override
     { return index == 0 ? dst_pd() : nullptr; }
 
-    virtual int n_inputs() const override { return 2; }
+    virtual int n_inputs() const override { return 3; }
     virtual int n_outputs() const override { return 1; }
 
     virtual status_t query(query_t what, int idx, void *result) const override