Align DLIE and nGraph average pooling layers

This commit is contained in:
Dmitry Kurtaev 2020-04-14 13:54:54 +03:00
parent 43ff8f8e8d
commit 870a775d7d

View File

@ -325,7 +325,7 @@ public:
ieLayer.setRoundingType(ceilMode ?
InferenceEngine::Builder::PoolingLayer::RoundingType::CEIL :
InferenceEngine::Builder::PoolingLayer::RoundingType::FLOOR);
ieLayer.setExcludePad(type == AVE && padMode == "SAME");
ieLayer.setExcludePad(!avePoolPaddedArea);
InferenceEngine::Builder::Layer l = ieLayer;
if (!padMode.empty())