From 18d9071be1bd0257af17516831a1dbf1910ca440 Mon Sep 17 00:00:00 2001 From: Vladimir Paramuzov Date: Fri, 7 May 2021 19:15:14 +0300 Subject: [PATCH] [IE CLDNN] Fixed FC weights check --- inference-engine/src/cldnn_engine/ops/matmul.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/inference-engine/src/cldnn_engine/ops/matmul.cpp b/inference-engine/src/cldnn_engine/ops/matmul.cpp index 289a5cb5d7df64..b18e199581cd15 100644 --- a/inference-engine/src/cldnn_engine/ops/matmul.cpp +++ b/inference-engine/src/cldnn_engine/ops/matmul.cpp @@ -60,8 +60,7 @@ void CreateMatMulOp(Program& p, const std::shared_ptr& o auto shape_a = op->get_input_shape(0); auto shape_b = op->get_input_shape(1); - bool is_fc = ngraph::is_type(op->get_input_node_shared_ptr(1)) || - ngraph::is_type(op->get_input_node_shared_ptr(1)); + bool is_fc = IsNodeOnConstPath(op->get_input_node_shared_ptr(1)); is_fc &= std::count_if(shape_b.begin(), shape_b.end(), [](size_t x) { return x != 1; }) <= 2; if (is_fc) {