diff options
Diffstat (limited to 'src/armnn/NetworkUtils.hpp')
| -rw-r--r-- | src/armnn/NetworkUtils.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/NetworkUtils.hpp b/src/armnn/NetworkUtils.hpp index 38fb22350d..064545aac5 100644 --- a/src/armnn/NetworkUtils.hpp +++ b/src/armnn/NetworkUtils.hpp @@ -11,6 +11,12 @@ namespace armnn { +std::vector<ConvertBf16ToFp32Layer*> InsertConvertBf16ToFp32LayersBefore(Graph& graph, + Layer& layer, + bool expectCorrectInputType = true); + +std::vector<ConvertFp32ToBf16Layer*> InsertConvertFp32ToBf16LayersAfter(Graph& graph, Layer& layer); + std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& graph, Layer& layer, bool expectCorrectInputType = true); |