src/armnn/test/TestUtils.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "TestUtils.hpp"
   7
   8 #include <boost/assert.hpp>
   9
  10 using namespace armnn;
  11
  12 void Connect(armnn::IConnectableLayer* from, armnn::IConnectableLayer* to, const armnn::TensorInfo& tensorInfo,
  13              unsigned int fromIndex, unsigned int toIndex)
  14 {
  15     BOOST_ASSERT(from);
  16     BOOST_ASSERT(to);
  17
  18     from->GetOutputSlot(fromIndex).Connect(to->GetInputSlot(toIndex));
  19     from->GetOutputSlot(fromIndex).SetTensorInfo(tensorInfo);
  20 }