diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-15 15:07:34 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
| commit | 974b65fea888cc000a5164d5b56d9ed016391151 (patch) | |
| tree | 406f1793d434e1b3ecdae997916361b18c6c6c69 /src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp | |
| parent | 39fedf04d226360a8c77ca1ca3e2528a709101b5 (diff) | |
| download | armnn-974b65fea888cc000a5164d5b56d9ed016391151.tar.gz | |
IVGCVSW-1951 Remove type templating from NeonConvolution2dWorkload
Change-Id: Id3d8137d60b14e93863ad5b6db752a7f6ef9b7fb
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp b/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp deleted file mode 100644 index 1cd30c70f9..0000000000 --- a/src/backends/neon/workloads/NeonConvolution2dBaseWorkload.hpp +++ /dev/null @@ -1,47 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include <backends/aclCommon/ArmComputeTensorUtils.hpp> -#include <backends/CpuTensorHandle.hpp> -#include <backends/neon/NeonLayerSupport.hpp> -#include <backends/neon/workloads/NeonWorkloadUtils.hpp> -#include <backends/Workload.hpp> - -#include <arm_compute/runtime/MemoryManagerOnDemand.h> - -#include <memory> - -namespace armnn -{ - -arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo& input, - const TensorInfo& output, - const Convolution2dDescriptor& descriptor, - const TensorInfo& weights, - const Optional<TensorInfo>& biases); - -template<armnn::DataType... dataTypes> -class NeonConvolution2dBaseWorkload : public TypedWorkload<Convolution2dQueueDescriptor, dataTypes...> -{ -public: - using TypedWorkload<Convolution2dQueueDescriptor, dataTypes...>::m_Data; - - NeonConvolution2dBaseWorkload(const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info, - std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); - - virtual void ValidateData() const {}; - -protected: - std::unique_ptr<arm_compute::IFunction> m_ConvolutionLayer; - - std::unique_ptr<arm_compute::Tensor> m_KernelTensor; - std::unique_ptr<arm_compute::Tensor> m_BiasTensor; - - void FreeUnusedTensors(); -}; - -} //namespace armnn |