ArmNN
 20.02
NetworkUtils.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "DeviceSpec.hpp"
9 #include "Graph.hpp"
10 
11 namespace armnn
12 {
13 
14 std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& graph,
15  Layer& layer,
16  bool expectCorrectInputType = true);
17 
18 std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer);
19 
20 std::vector<DebugLayer*> InsertDebugLayerAfter(Graph& graph, Layer& layer);
21 
22 } // namespace armnn
std::vector< ConvertFp32ToFp16Layer * > InsertConvertFp32ToFp16LayersAfter(Graph &graph, Layer &layer)
std::vector< ConvertFp16ToFp32Layer * > InsertConvertFp16ToFp32LayersBefore(Graph &graph, Layer &layer, bool expectCorrectInputType)
Copyright (c) 2020 ARM Limited.
std::vector< DebugLayer * > InsertDebugLayerAfter(Graph &graph, Layer &layer)