1 // 2 // Copyright © 2022 Arm Ltd and Contributors. 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, bool toFile); 21 22 bool RevertConstantWeightsToFP32(Layer* layer); 23 24 } // namespace armnn 25