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<ConvertBf16ToFp32Layer*> InsertConvertBf16ToFp32LayersBefore(Graph& graph, 15 Layer& layer, 16 bool expectCorrectInputType = true); 17 18 std::vector<ConvertFp32ToBf16Layer*> InsertConvertFp32ToBf16LayersBefore(Graph& graph, 19 Layer& layer, 20 bool expectCorrectInputType = true); 21 22 std::vector<ConvertFp32ToBf16Layer*> InsertConvertFp32ToBf16LayersAfter(Graph& graph, Layer& layer); 23 24 std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& graph, 25 Layer& layer, 26 bool expectCorrectInputType = true); 27 28 std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer); 29 30 std::vector<DebugLayer*> InsertDebugLayerAfter(Graph& graph, Layer& layer); 31 32 } // namespace armnn 33