aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/TestUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/TestUtils.cpp')
-rw-r--r--src/armnn/test/TestUtils.cpp20
1 files changed, 20 insertions, 0 deletions
diff --git a/src/armnn/test/TestUtils.cpp b/src/armnn/test/TestUtils.cpp
new file mode 100644
index 0000000000..b0ed17e41a
--- /dev/null
+++ b/src/armnn/test/TestUtils.cpp
@@ -0,0 +1,20 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "TestUtils.hpp"
+
+#include <boost/assert.hpp>
+
+using namespace armnn;
+
+void Connect(armnn::IConnectableLayer* from, armnn::IConnectableLayer* to, const armnn::TensorInfo& tensorInfo,
+ unsigned int fromIndex, unsigned int toIndex)
+{
+ BOOST_ASSERT(from);
+ BOOST_ASSERT(to);
+
+ from->GetOutputSlot(fromIndex).Connect(to->GetInputSlot(toIndex));
+ from->GetOutputSlot(fromIndex).SetTensorInfo(tensorInfo);
+}