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,
23 const ActivationDescriptor* activationDescriptor =
nullptr);
30 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
31 const arm_compute::CLCompileContext& clCompileContext);
37 mutable arm_compute::CLFullyConnectedLayer m_FullyConnectedLayer;
39 std::unique_ptr<arm_compute::CLTensor> m_WeightsTensor;
40 std::unique_ptr<arm_compute::CLTensor> m_BiasesTensor;
42 void FreeUnusedTensors();
arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const TensorInfo &weights, const TensorInfo &biases, const FullyConnectedDescriptor &descriptor, const ActivationDescriptor *activationDescriptor)
Copyright (c) 2021 ARM Limited and Contributors.
ClFullyConnectedWorkload(const armnn::FullyConnectedQueueDescriptor &descriptor, const armnn::WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override
Contains information about TensorInfos of a layer.