IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GNA] Remove FQ layers positioned after a memory layer (#5323)

Browse Source
This commit is contained in:
Szymon Irzabek 2021-04-26 10:44:18 +02:00 committed by GitHub
parent d2868c9873
commit af6f315071
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/src/gna_plugin/optimizer/gna_pass_manager.cpp
View File

@ -1994,7 +1994,7 @@ void MoveFakeQuantizeLayerIntoQuantParamsPass :: run() {
};
auto prevLayer = CNNNetPrevLayerSkipCertain(layer, 0, skipNonFunctional);
if (LayerInfo(prevLayer).isActivation() || LayerInfo(prevLayer).isConst()) {
if (LayerInfo(prevLayer).isActivation() || LayerInfo(prevLayer).isConst() || LayerInfo(prevLayer).isMemory()) {
return true;
}
@ -2119,7 +2119,7 @@ void MoveFakeQuantizeLayerIntoQuantParamsPass :: run() {
}
// Allow FQ Fuse checks if FQ layer can be fused to a layer before or after.
// FQ Layer is fused only when previous layer is const or activation layer
// FQ Layer is fused only when previous layer is const, memory or activation layer
// or a next layer is activation layer.
bool isFQFuseAllowed = allowFQFuse(l);
auto prevData = prevLayer->outData.front();