diff options
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h index 82969301b0..7767b73e10 100644 --- a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h @@ -94,12 +94,14 @@ public: // Inherited methods overridden: void run() override; + void prepare() override; private: CLMemoryGroup _memory_group; CLDeconvolutionLayerUpsample _scale_f; CLConvolutionLayer _conv_f; CLTensor _scaled_output; + bool _is_prepared; }; } #endif /* __ARM_COMPUTE_CLDECONVOLUTIONLAYER_H__ */ |