[LPT] GemmTransformation usage fix (#1899)

This commit is contained in:
Edward Shogulin 2020-08-24 20:09:01 +03:00 committed by GitHub
parent 4013649fe8
commit db2e5c0728
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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();