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