diff options
author | Ryan OShea <Ryan.OShea2@arm.com> | 2020-04-22 16:37:29 +0100 |
---|---|---|
committer | Ryan OShea <Ryan.OShea2@arm.com> | 2020-04-23 14:28:20 +0100 |
commit | ed27ee7e1629b8af6f32ad0aed0901d749b53c32 (patch) | |
tree | 0fd7bee16c8e47ec0a4373476695aa0fe66157c9 /CMakeLists.txt | |
parent | f7b9d29c69e707195df8e37f429f9bff521fd490 (diff) | |
download | armnn-ed27ee7e1629b8af6f32ad0aed0901d749b53c32.tar.gz |
IVGCVSW-4496 Add Flatten support to ONNX parser
* Added ParseFlatten method
* Added Read int64 attribute method
- Modified ComputeReshapeInfo method
- Modified ParseReshape
* Added unit tests
- Reorganised OnnxParser.cpp/.hpp
Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I8a9553438dd1e8c702d821b093587e0074c027d5
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 9e31a0392f..20e8717c82 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -820,20 +820,21 @@ if(BUILD_UNIT_TESTS) if(BUILD_ONNX_PARSER AND ARMNNREF) list(APPEND unittest_sources - src/armnnOnnxParser/test/Constructor.cpp - src/armnnOnnxParser/test/CreateNetwork.cpp - src/armnnOnnxParser/test/ProtoxtFixture.cpp + src/armnnOnnxParser/test/Addition.cpp + src/armnnOnnxParser/test/BatchNorm.cpp src/armnnOnnxParser/test/Clip.cpp src/armnnOnnxParser/test/Const.cpp - src/armnnOnnxParser/test/Pooling.cpp - src/armnnOnnxParser/test/Reshape.cpp - src/armnnOnnxParser/test/Relu.cpp + src/armnnOnnxParser/test/Constructor.cpp src/armnnOnnxParser/test/Conv2D.cpp - src/armnnOnnxParser/test/Addition.cpp + src/armnnOnnxParser/test/CreateNetwork.cpp + src/armnnOnnxParser/test/DepthConv.cpp + src/armnnOnnxParser/test/Flatten.cpp src/armnnOnnxParser/test/FullyConnected.cpp src/armnnOnnxParser/test/GetInputsOutputs.cpp - src/armnnOnnxParser/test/BatchNorm.cpp - src/armnnOnnxParser/test/DepthConv.cpp + src/armnnOnnxParser/test/Pooling.cpp + src/armnnOnnxParser/test/ProtoxtFixture.cpp + src/armnnOnnxParser/test/Relu.cpp + src/armnnOnnxParser/test/Reshape.cpp ) endif() |