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

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