diff options
| author | Keith Davis <keith.davis@arm.com> | 2022-10-14 15:50:33 +0100 |
|---|---|---|
| committer | KeithARM <keith.davis@arm.com> | 2022-10-19 10:33:40 +0000 |
| commit | 15f9c68adef324cd0158cea3d021c0f6bef5eecf (patch) | |
| tree | 1cd48b345d182fd19efdc40a32e2540befd8f925 /tests/InferenceModel.hpp | |
| parent | 7bbf56598010041ea46c3fa9d32604db777ee26e (diff) | |
| download | armnn-15f9c68adef324cd0158cea3d021c0f6bef5eecf.tar.gz | |
MLCE-545 INT8 TFLite model execution abnormal
* Add functionality to print output tensors to file in tempdir
* UnitTests
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Idfb4c186544187db1fecdfca11c662540f645439
Diffstat (limited to 'tests/InferenceModel.hpp')
| -rw-r--r-- | tests/InferenceModel.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 268f60301c..d837fc1fcf 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -68,6 +68,7 @@ struct Params bool m_EnableFp16TurboMode; bool m_EnableBf16TurboMode; bool m_PrintIntermediateLayers; + bool m_PrintIntermediateLayersToFile; bool m_ParseUnsupported; bool m_InferOutputShape; bool m_EnableFastMath; @@ -91,6 +92,7 @@ struct Params , m_EnableFp16TurboMode(false) , m_EnableBf16TurboMode(false) , m_PrintIntermediateLayers(false) + , m_PrintIntermediateLayersToFile(false) , m_ParseUnsupported(false) , m_InferOutputShape(false) , m_EnableFastMath(false) @@ -452,6 +454,7 @@ public: options.m_ReduceFp32ToFp16 = params.m_EnableFp16TurboMode; options.m_ReduceFp32ToBf16 = params.m_EnableBf16TurboMode; options.m_Debug = params.m_PrintIntermediateLayers; + options.m_DebugToFile = params.m_PrintIntermediateLayersToFile; options.m_shapeInferenceMethod = params.m_InferOutputShape ? armnn::ShapeInferenceMethod::InferAndValidate : armnn::ShapeInferenceMethod::ValidateOnly; options.m_ProfilingEnabled = m_EnableProfiling; |