diff options
Diffstat (limited to 'src/armnn/NetworkUtils.hpp')
| -rw-r--r-- | src/armnn/NetworkUtils.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/NetworkUtils.hpp b/src/armnn/NetworkUtils.hpp index 421c52a6a7..38fb22350d 100644 --- a/src/armnn/NetworkUtils.hpp +++ b/src/armnn/NetworkUtils.hpp @@ -11,7 +11,9 @@ namespace armnn { -std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& graph, Layer& layer); +std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& graph, + Layer& layer, + bool expectCorrectInputType = true); std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer); |