From ed27ee7e1629b8af6f32ad0aed0901d749b53c32 Mon Sep 17 00:00:00 2001 From: Ryan OShea Date: Wed, 22 Apr 2020 16:37:29 +0100 Subject: 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 Change-Id: I8a9553438dd1e8c702d821b093587e0074c027d5 --- CMakeLists.txt | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) (limited to 'CMakeLists.txt') 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() -- cgit v1.2.1