diff options
Diffstat (limited to 'src/armnn/test/TestUtils.cpp')
-rw-r--r-- | src/armnn/test/TestUtils.cpp | 20 |
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); +} |