diff options
| -rw-r--r-- | src/armnn/optimizations/FoldPadIntoLayer2d.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/optimizations/FoldPadIntoLayer2d.hpp b/src/armnn/optimizations/FoldPadIntoLayer2d.hpp index 5592491653..e88e4980db 100644 --- a/src/armnn/optimizations/FoldPadIntoLayer2d.hpp +++ b/src/armnn/optimizations/FoldPadIntoLayer2d.hpp @@ -157,7 +157,7 @@ Layer2dT* FoldPadIntoLayer2dImpl(Graph& graph, InputSlot& connection) // Workaround an issue in the compute library. The conv2d algorithm that the // compute library is choosing is not handling the 1x1 filter case when // the padding size >= filter size - if constexpr (std::is_same<Layer2dT, armnn::Convolution2dLayer>::value) + if (layer2d.GetType() == armnn::LayerType::Convolution2d) { // Get filter width and height armnnUtils::DataLayoutIndexed dataLayoutIndex(newLayer2dDescriptor.m_DataLayout); |