Merge pull request #17060 from dkurt:dnn_align_ie_pool

This commit is contained in:
Alexander Alekhin 2020-04-14 15:15:44 +00:00
commit f6de25b6cb

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