diff options
| author | Cathal Corbett <catcor01@e127348.nice.arm.com> | 2021-10-07 11:46:40 +0100 |
|---|---|---|
| committer | Cathal Corbett <cathal.corbett@arm.com> | 2021-10-08 11:28:35 +0000 |
| commit | 521032fd424cf86681eb125afbf5eaee47d8c585 (patch) | |
| tree | 65162778f203638f1c039097b8240422f99dad76 /src/backends/reference/test/RefEndToEndTests.cpp | |
| parent | 723bc3b5d8a911a369eee658631d9f107ea09896 (diff) | |
| download | armnn-521032fd424cf86681eb125afbf5eaee47d8c585.tar.gz | |
IVGCVSW-6417: Catch AddFullyConnected API error when weights TensorInfo isn't set
* Updated code in Graph.cpp InferTensorInfos() to be more descriptive.
* Added method VerifyConstantLayerSetTensorInfo() in Graph.cpp/hpp
to error when ConstantLayer TensorInfo is not set.
* Updated Optimize() in Network.cpp to call VerifyConstantLayerSetTensorInfo().
* Added unit test with ConstantLayer TensorInfo not
set to catch error in VerifyConstantLayerSetTensorInfo().
* Added comments around method VerifyConstantLayerSetTensorInfo().
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I366596243f7c5823676222e2d0cce1335bc8c325
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index ed4b229ace..6c11a75e96 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -618,12 +618,17 @@ TEST_CASE("RefFullyConnectedEndToEndTestFloat32") TEST_CASE("RefFullyConnectedEndToEndTestNonConstantWeightsConstantBiasesFloat32") { - FullyConnectedWithDynamicOrConstantInputsEndToEnd<armnn::DataType::Float32>(defaultBackends, true, true); + FullyConnectedWithDynamicOrConstantInputsEndToEnd<armnn::DataType::Float32>(defaultBackends, true, true, true); } TEST_CASE("RefFullyConnectedEndToEndTestConstantWeightsNonConstantBiasesFloat32") { - FullyConnectedWithDynamicOrConstantInputsEndToEnd<armnn::DataType::Float32>(defaultBackends, true, false); + FullyConnectedWithDynamicOrConstantInputsEndToEnd<armnn::DataType::Float32>(defaultBackends, true, false, true); +} + +TEST_CASE("RefFullyConnectedEndToEndTestConstantWeightsTensorInfoNotSet") +{ + FullyConnectedWithDynamicOrConstantInputsEndToEnd<armnn::DataType::Float32>(defaultBackends, true, false, false); } TEST_CASE("RefGatherFloatTest") |