diff options
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp index c7b082183c..eab24810d5 100644 --- a/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp @@ -14,7 +14,7 @@ std::unique_ptr<armnn::IWorkload> CreateWorkload<armnn::MultiplicationQueueDescr const armnn::WorkloadInfo& info, const armnn::MultiplicationQueueDescriptor& descriptor) { - return workloadFactory.CreateMultiplication(descriptor, info); + return workloadFactory.CreateWorkload(armnn::LayerType::Multiplication, descriptor, info); } LayerTestResult<float, 4> MultiplicationTest(armnn::IWorkloadFactory& workloadFactory, @@ -571,8 +571,10 @@ LayerTestResult<float,4> CompareMultiplicationTest( SetWorkloadInput(refData, refInfo, 1, inputTensorInfo1, inputHandle1Ref.get()); SetWorkloadOutput(refData, refInfo, 0, outputTensorInfo, outputHandleRef.get()); - std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateMultiplication(data, info); - std::unique_ptr<armnn::IWorkload> workloadRef = refWorkloadFactory.CreateMultiplication(refData, refInfo); + std::unique_ptr<armnn::IWorkload> workload + = workloadFactory.CreateWorkload(armnn::LayerType::Multiplication, data, info); + std::unique_ptr<armnn::IWorkload> workloadRef + = refWorkloadFactory.CreateWorkload(armnn::LayerType::Multiplication, refData, refInfo); inputHandle0->Allocate(); inputHandle1->Allocate(); |