10 #include <arm_compute/runtime/IFunction.h> 11 #include <arm_compute/runtime/Tensor.h> 12 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 14 #include <arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h> 22 const TensorInfo& output,
23 const TransposeConvolution2dDescriptor& descriptor,
24 const TensorInfo& weights,
25 const Optional<TensorInfo>& biases);
32 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
37 std::unique_ptr<arm_compute::NEDeconvolutionLayer> m_Layer;
39 std::unique_ptr<arm_compute::Tensor> m_KernelTensor;
40 std::unique_ptr<arm_compute::Tensor> m_BiasTensor;
42 void FreeUnusedTensors();
arm_compute::Status NeonTransposeConvolution2dWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const TransposeConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases)
void Execute() const override
NeonTransposeConvolution2dWorkload(const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)