diff options
Diffstat (limited to 'arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h')
-rw-r--r-- | arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h b/arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h index 6f70d3c3a0..14474f4ee5 100644 --- a/arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h +++ b/arm_compute/graph/backends/FusedDepthwiseConvolutionBatchNormalizationFunction.h @@ -42,8 +42,8 @@ public: using TensorType = typename TargetInfo::TensorType; using TensorConcreteType = typename TargetInfo::TensorConcreteType; - FusedDepthwiseConvolutionBatchNormalizationFunction() - : _depth_conv_layer(), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) + FusedDepthwiseConvolutionBatchNormalizationFunction(std::shared_ptr<IMemoryManager> memory_manager = nullptr) + : _depth_conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false) { } |