From 521032fd424cf86681eb125afbf5eaee47d8c585 Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Thu, 7 Oct 2021 11:46:40 +0100 Subject: 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 Change-Id: I366596243f7c5823676222e2d0cce1335bc8c325 --- src/backends/reference/test/RefEndToEndTests.cpp | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'src/backends/reference') 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(defaultBackends, true, true); + FullyConnectedWithDynamicOrConstantInputsEndToEnd(defaultBackends, true, true, true); } TEST_CASE("RefFullyConnectedEndToEndTestConstantWeightsNonConstantBiasesFloat32") { - FullyConnectedWithDynamicOrConstantInputsEndToEnd(defaultBackends, true, false); + FullyConnectedWithDynamicOrConstantInputsEndToEnd(defaultBackends, true, false, true); +} + +TEST_CASE("RefFullyConnectedEndToEndTestConstantWeightsTensorInfoNotSet") +{ + FullyConnectedWithDynamicOrConstantInputsEndToEnd(defaultBackends, true, false, false); } TEST_CASE("RefGatherFloatTest") -- cgit v1.2.1