diff options
Diffstat (limited to 'src/armnn/layers/DepthwiseConvolution2dLayer.cpp')
| -rw-r--r-- | src/armnn/layers/DepthwiseConvolution2dLayer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/DepthwiseConvolution2dLayer.cpp b/src/armnn/layers/DepthwiseConvolution2dLayer.cpp index dc6b2c2fe7..7efb307091 100644 --- a/src/armnn/layers/DepthwiseConvolution2dLayer.cpp +++ b/src/armnn/layers/DepthwiseConvolution2dLayer.cpp @@ -119,8 +119,10 @@ DepthwiseConvolution2dLayer::InferOutputShapes(const std::vector<TensorShape>& i return std::vector<TensorShape>{ tensorShape }; } -void DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs() +void DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(ShapeInferenceMethod shapeInferenceMethod) { + IgnoreUnused(shapeInferenceMethod); + VerifyLayerConnections(1, CHECK_LOCATION()); // on this level constant data should not be released.. |