10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/IFunction.h> 12 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 13 #include <arm_compute/runtime/Tensor.h> 21 const TensorInfo& output,
22 const TensorInfo& weights,
23 const TensorInfo& biases,
24 const FullyConnectedDescriptor& descriptor,
25 const ActivationDescriptor* activationDescriptor =
nullptr);
31 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager);
32 virtual void Execute()
const override;
35 std::unique_ptr<arm_compute::IFunction> m_FullyConnectedLayer;
37 std::unique_ptr<arm_compute::Tensor> m_WeightsTensor;
38 std::unique_ptr<arm_compute::Tensor> m_BiasesTensor;
40 void FreeUnusedTensors();
NeonFullyConnectedWorkload(const FullyConnectedQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)
arm_compute::Status NeonFullyConnectedWorkloadValidate(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.
Contains information about TensorInfos of a layer.
virtual void Execute() const override