ArmNN
NotReleased
TestUtils.cpp
Go to the documentation of this file.
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
}
armnn::IConnectableLayer::GetInputSlot
virtual const IInputSlot & GetInputSlot(unsigned int index) const =0
armnn::TensorInfo
Definition:
Tensor.hpp:53
armnn::IOutputSlot::SetTensorInfo
virtual void SetTensorInfo(const TensorInfo &tensorInfo)=0
armnn
Definition:
BackendHelper.hpp:11
Connect
void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)
Definition:
TestUtils.cpp:12
armnn::IConnectableLayer
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
Definition:
INetwork.hpp:61
TestUtils.hpp
armnn::IOutputSlot::Connect
virtual int Connect(IInputSlot &destination)=0
armnn::IConnectableLayer::GetOutputSlot
virtual const IOutputSlot & GetOutputSlot(unsigned int index) const =0
src
armnn
test
TestUtils.cpp
Generated on Fri Mar 13 2020 16:06:55 for ArmNN by
1.8.13