diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-10 12:32:08 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-10 14:58:18 +0000 |
commit | 1a763dd350e80f26a661dc5216fdef1256e5b32a (patch) | |
tree | 866c985af88766b2f08f317eed12cbf37b618362 /src/armnn/test/CreateWorkload.hpp | |
parent | 0dd3b43e1497ab406acfdefca425143fbefb1ee4 (diff) | |
download | armnn-1a763dd350e80f26a661dc5216fdef1256e5b32a.tar.gz |
IVGCVSW-3829 Add CL workload for RSQRT
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Idf40ed0ebf2af3170137bf3a3cb5a9635a203d4a
Diffstat (limited to 'src/armnn/test/CreateWorkload.hpp')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index b50135aea5..d6d1ab3c5b 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -129,6 +129,32 @@ std::unique_ptr<WorkloadType> CreateElementwiseWorkloadTest(armnn::IWorkloadFact return workload; } +template <typename WorkloadType, + typename DescriptorType, + typename LayerType, + armnn::DataType DataType> +std::unique_ptr<WorkloadType> CreateElementwiseUnaryWorkloadTest(armnn::IWorkloadFactory & factory, + armnn::Graph & graph) +{ + Layer* const layer = graph.AddLayer<LayerType>("layer"); + + Layer* const input = graph.AddLayer<InputLayer>(0, "input"); + Layer* const output = graph.AddLayer<OutputLayer>(0, "output"); + + armnn::TensorInfo tensorInfo({ 2, 3 }, DataType); + Connect(input, layer, tensorInfo, 0, 0); + Connect(layer, output, tensorInfo, 0, 0); + CreateTensorHandles(graph, factory); + + auto workload = MakeAndCheckWorkload<WorkloadType>(*layer, graph, factory); + DescriptorType queueDescriptor = workload->GetData(); + + BOOST_TEST(queueDescriptor.m_Inputs.size() == 1); + BOOST_TEST(queueDescriptor.m_Outputs.size() == 1); + + return workload; +} + template <typename BatchNormalizationWorkloadType, armnn::DataType DataType> std::unique_ptr<BatchNormalizationWorkloadType> CreateBatchNormalizationWorkloadTest( armnn::IWorkloadFactory& factory, armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW) |