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();
arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor)
Copyright (c) 2020 ARM Limited.
void Execute() const override
ClFullyConnectedWorkload(const armnn::FullyConnectedQueueDescriptor &descriptor, const armnn::WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
Contains information about inputs and outputs to a layer.