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