diff options
Diffstat (limited to 'src/armnn/optimizations/RedirectMembersToConstantInputs.hpp')
| -rw-r--r-- | src/armnn/optimizations/RedirectMembersToConstantInputs.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp b/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp index 85d715c6b1..cb97a0fe32 100644 --- a/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp +++ b/src/armnn/optimizations/RedirectMembersToConstantInputs.hpp @@ -31,6 +31,7 @@ public: case LayerType::Convolution2d: break; case LayerType::DepthwiseConvolution2d: + RedirectWeightsAndBiases<DepthwiseConvolution2dLayer>(&layer); break; case LayerType::DetectionPostProcess: break; @@ -80,7 +81,7 @@ private: } }; -using RedirectMembersToConstantInputs = OptimizeForType<FullyConnectedLayer, RedirectMembersToConstantInputsImpl>; +using RedirectMembersToConstantInputs = OptimizeForType<Layer, RedirectMembersToConstantInputsImpl>; } // namespace optimizations } // namespace armnn |