diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp')
| -rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp b/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp index 9cf272e9f5..aca0ba5337 100644 --- a/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonDepthwiseConvolutionUint8Workload.hpp @@ -20,8 +20,10 @@ public: private: mutable std::unique_ptr<arm_compute::IFunction> m_pDepthwiseConvolutionLayer; - arm_compute::Tensor m_KernelTensor; - arm_compute::Tensor m_BiasTensor; + std::unique_ptr<arm_compute::Tensor> m_KernelTensor; + std::unique_ptr<arm_compute::Tensor> m_BiasTensor; + + void FreeUnusedTensors(); }; } //namespace armnn |