From 5b2145c92dabb68a0ec7ff65948f52d3fdcecf4a Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Mon, 19 Dec 2022 13:36:36 +0000 Subject: IVGCVSW-6632 Return INetwork* not INetworkPtr& from OptimizationViews::GetINetwork() * Also provide API to return reference via GetINetworkRef(). Signed-off-by: Francis Murtagh Change-Id: I80f4912cb7e22a2023e9bb880182ad4610c219e9 --- src/backends/backendsCommon/test/OptimizationViewsTests.cpp | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) (limited to 'src/backends/backendsCommon') diff --git a/src/backends/backendsCommon/test/OptimizationViewsTests.cpp b/src/backends/backendsCommon/test/OptimizationViewsTests.cpp index 9b86784dce..ff84eea2de 100644 --- a/src/backends/backendsCommon/test/OptimizationViewsTests.cpp +++ b/src/backends/backendsCommon/test/OptimizationViewsTests.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017, 2019-2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -288,4 +288,15 @@ TEST_CASE("OptimizeViewsValidateDeviceMockBackend") CheckLayers(graph); } +TEST_CASE("OptimizedViewsReturnsINetworkReference") +{ + OptimizationViews view; + + auto layer = view.GetINetworkRef().AddInputLayer(0, "input"); + + // Check layer has been added to the referenced INetwork + CHECK(layer); +} + + } \ No newline at end of file -- cgit v1.2.1