diff options
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp index 2640c329b3..016d27897e 100644 --- a/src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/ComparisonTestImpl.cpp @@ -66,7 +66,8 @@ LayerTestResult<uint8_t, NumDims> ComparisonTestImpl( AddInputToWorkload(qDescriptor, info, inputTensorInfo1, inputHandle1.get()); AddOutputToWorkload(qDescriptor, info, outputTensorInfo, outputHandle.get()); - std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateComparison(qDescriptor, info); + std::unique_ptr<armnn::IWorkload> workload + = workloadFactory.CreateWorkload(armnn::LayerType::Comparison, qDescriptor, info); inputHandle0->Allocate(); inputHandle1->Allocate(); |