[LPT] GemmTransformation usage fix (openvinotoolkit#1899)

RomanZm · Aug 28, 2020 · 7888ef7 · 7888ef7
1 parent 1fc8cfb
commit 7888ef7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/inference-engine/src/cldnn_engine/cldnn_program.cpp b/inference-engine/src/cldnn_engine/cldnn_program.cpp
@@ -88,9 +88,8 @@
 #include <exec_graph_info.hpp>
 
 #include "low_precision_transformations/transformer.hpp"
-#include "low_precision_transformations/eltwise.hpp"
-#include "low_precision_transformations/concat_multi_channels.hpp"
 #include "low_precision_transformations/fully_connected.hpp"
+#include "low_precision_transformations/gemm.hpp"
 
 #include <iostream>
 #include <iomanip>
@@ -241,7 +240,7 @@ Program::Program(InferenceEngine::ICNNNetwork& network, std::shared_ptr<const cl
 
         auto transforms = LowPrecisionTransformer::getAllTransformations(params)
                 .add<FullyConnectedTransformation>(LayerTransformation::Params(params).setSupportAsymmetricQuantization(false), "FullyConnected")
-                .add<FullyConnectedTransformation>(LayerTransformation::Params(params).setSupportAsymmetricQuantization(false), "GEMM");
+                .add<GemmTransformation>(LayerTransformation::Params(params).setSupportAsymmetricQuantization(false), "GEMM");
 
         auto it = details::CNNNetworkIterator(&network);
         auto end = details::CNNNetworkIterator();