diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-17 11:17:41 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
commit | ab8cdc13443408d727fd38be42315cf942251940 (patch) | |
tree | f1576377150b72f1a2404915adcf1affd5eab3e4 /src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp | |
parent | ca225f0ab9aac74ccc7c62cfcf46c95f7715b2ee (diff) | |
download | armnn-ab8cdc13443408d727fd38be42315cf942251940.tar.gz |
IVGCVSW-949 Add 8-bit fully connected support
Change-Id: I0953bb8dbc4b76001f207e37c8c2742a6ebd888b
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp | 42 |
1 files changed, 0 insertions, 42 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp b/src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp deleted file mode 100644 index e8d6a7897d..0000000000 --- a/src/armnn/backends/ClWorkloads/ClFullyConnectedFloatWorkload.hpp +++ /dev/null @@ -1,42 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include "backends/ClWorkloadUtils.hpp" - -#include "arm_compute/runtime/MemoryManagerOnDemand.h" - -#include <memory> - -namespace armnn -{ - -arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo& input, - const TensorInfo& output, - const TensorInfo& weights, - const TensorInfo& biases, - const FullyConnectedDescriptor& descriptor); - -class ClFullyConnectedFloatWorkload : public armnn::FloatWorkload<armnn::FullyConnectedQueueDescriptor> -{ -public: - ClFullyConnectedFloatWorkload(const armnn::FullyConnectedQueueDescriptor& descriptor, - const armnn::WorkloadInfo& info, - std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); - - using armnn::FloatWorkload<armnn::FullyConnectedQueueDescriptor>::m_Data; - void Execute() const override; - -private: - mutable arm_compute::CLFullyConnectedLayer m_FullyConnectedLayer; - - std::unique_ptr<arm_compute::CLTensor> m_WeightsTensor; - std::unique_ptr<arm_compute::CLTensor> m_BiasesTensor; - - void FreeUnusedTensors(); -}; - -} //namespace armnn |