ArmNN
NotReleased
GraphUtils.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include <
Graph.hpp
>
8
9
#include <string>
10
11
12
bool
GraphHasNamedLayer
(
const
armnn::Graph
& graph,
const
std::string& name);
13
14
armnn::Layer
*
GetFirstLayerWithName
(
armnn::Graph
& graph,
const
std::string& name);
15
16
bool
CheckNumberOfInputSlot
(
armnn::Layer
* layer,
unsigned
int
num);
17
18
bool
CheckNumberOfOutputSlot
(
armnn::Layer
* layer,
unsigned
int
num);
19
20
bool
IsConnected
(
armnn::Layer
* srcLayer,
armnn::Layer
* destLayer,
21
unsigned
int
srcSlot,
unsigned
int
destSlot,
22
const
armnn::TensorInfo
& expectedTensorInfo);
23
CheckNumberOfOutputSlot
bool CheckNumberOfOutputSlot(armnn::Layer *layer, unsigned int num)
Definition:
GraphUtils.cpp:37
IsConnected
bool IsConnected(armnn::Layer *srcLayer, armnn::Layer *destLayer, unsigned int srcSlot, unsigned int destSlot, const armnn::TensorInfo &expectedTensorInfo)
Definition:
GraphUtils.cpp:42
armnn::Layer
Definition:
Layer.hpp:209
armnn::TensorInfo
Definition:
Tensor.hpp:53
CheckNumberOfInputSlot
bool CheckNumberOfInputSlot(armnn::Layer *layer, unsigned int num)
Definition:
GraphUtils.cpp:32
GetFirstLayerWithName
armnn::Layer * GetFirstLayerWithName(armnn::Graph &graph, const std::string &name)
Definition:
GraphUtils.cpp:20
GraphHasNamedLayer
bool GraphHasNamedLayer(const armnn::Graph &graph, const std::string &name)
Definition:
GraphUtils.cpp:8
armnn::Graph
Definition:
Graph.hpp:29
Graph.hpp
src
armnn
test
GraphUtils.hpp
Generated on Fri Mar 13 2020 16:06:54 for ArmNN by
1.8.13