10 #include <arm_compute/runtime/IFunction.h> 11 #include <arm_compute/runtime/Tensor.h> 19 const TensorInfo& output,
20 const DepthwiseConvolution2dDescriptor& descriptor,
21 const TensorInfo& weights,
22 const Optional<TensorInfo>& biases);
30 virtual void Execute()
const override;
33 mutable std::unique_ptr<arm_compute::IFunction> m_pDepthwiseConvolutionLayer;
35 std::unique_ptr<arm_compute::Tensor> m_KernelTensor;
36 std::unique_ptr<arm_compute::Tensor> m_BiasTensor;
38 void FreeUnusedTensors();
Copyright (c) 2020 ARM Limited.
virtual void Execute() const override
arm_compute::Status NeonDepthwiseConvolutionWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const DepthwiseConvolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases)
Contains information about inputs and outputs to a layer.
NeonDepthwiseConvolutionWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)