diff options
Diffstat (limited to 'src/backends/reference')
| -rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 2 | ||||
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 15 |
2 files changed, 14 insertions, 3 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index 086f8eea8d..76c18a4fd6 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -279,7 +279,7 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreateWorkload(LayerType type, { return std::make_unique<RefDebugBooleanWorkload>(*debugQueueDescriptor, info); } - return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(*debugQueueDescriptor, info); + return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(*debugQueueDescriptor, info); } case LayerType::DepthToSpace: { diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 73786b5ccd..6f57236dd5 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -626,6 +626,13 @@ TEST_CASE("RefConvolution2dFloat16Test") Convolution2dEndToEnd<armnn::DataType::Float16>(defaultBackends, armnn::DataLayout::NHWC); } +TEST_CASE("RefConvolution2dInt8Test") +{ + Convolution2dEndToEnd<armnn::DataType::QSymmS8, + armnn::DataType::QSymmS8, + armnn::DataType::Signed32>(defaultBackends, armnn::DataLayout::NHWC); +} + TEST_CASE("RefConvolution3dFloat32Test") { Convolution3dEndToEnd<armnn::DataType::Float32, armnn::DataType::Float32>(defaultBackends, @@ -818,6 +825,12 @@ TEST_CASE("DepthToSpaceEndToEndNhwcInt16") DepthToSpaceEndToEnd<armnn::DataType::QSymmS16>(defaultBackends, armnn::DataLayout::NHWC); } +// Quantization +TEST_CASE("RefQuantizeInt8") +{ + QuantizationEndToEndInt8(defaultBackends); +} + // Dequantize TEST_CASE("DequantizeEndToEndSimpleTest") { @@ -1341,7 +1354,6 @@ TEST_CASE("QuantizationEndToEndFloat16_S16Test") } // ScatterNd - TEST_CASE("RefScatterNd1DInputEndToEndFloat32Test") { ScatterNd1DimUpdateWithInputEndToEnd<armnn::DataType::Float32>(defaultBackends); @@ -1382,7 +1394,6 @@ TEST_CASE("RefScatterNd2DNoInputEndToEndInt8Test") ScatterNd2DimUpdateNoInputEndToEnd<armnn::DataType::QAsymmS8>(defaultBackends); } - // SpaceToDepth TEST_CASE("RefSpaceToDepthNhwcEndToEndTest1") { |