10 #include <arm_compute/runtime/CL/functions/CLFullyConnectedLayer.h> 11 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 19 const TensorInfo& output,
20 const TensorInfo& weights,
21 const TensorInfo& biases,
22 const FullyConnectedDescriptor& descriptor);
29 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
35 mutable arm_compute::CLFullyConnectedLayer m_FullyConnectedLayer;
37 std::unique_ptr<arm_compute::CLTensor> m_WeightsTensor;
38 std::unique_ptr<arm_compute::CLTensor> m_BiasesTensor;
40 void FreeUnusedTensors();
void Execute() const override
arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor)
ClFullyConnectedWorkload(const armnn::FullyConnectedQueueDescriptor &descriptor, const armnn::WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)