From 44a7167e0f13dc1d703cd573f57636fde711c618 Mon Sep 17 00:00:00 2001
From: Matteo Martincigh <matteo.martincigh@arm.com>
Date: Tue, 11 Dec 2018 13:46:52 +0000
Subject: Fix the ONNX parser tests issues due to a change in the latest ONNx
 master

 * A recent change in ONNX now forces the data types to be indicated by the
   integer corresponding to their original enum value
 * The new values are taken from the TensorProto_DataType enum at onnx.pb.h:153

Change-Id: I6054611795a7edb11463396eaefc4dd7e27261a0
---
 src/armnnOnnxParser/test/Const.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/armnnOnnxParser/test/Const.cpp')

diff --git a/src/armnnOnnxParser/test/Const.cpp b/src/armnnOnnxParser/test/Const.cpp
index 130b9263e8..8777757a96 100644
--- a/src/armnnOnnxParser/test/Const.cpp
+++ b/src/armnnOnnxParser/test/Const.cpp
@@ -37,7 +37,7 @@ struct ConstMainFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxPars
                               float_data: 6.0
 
                           }
-                          type: FLOAT
+                          type: 1
                         }
                         name:  "constantNode"
                         op_type:  "Constant"
@@ -46,7 +46,7 @@ struct ConstMainFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxPars
                           name:  "Output"
                           type {
                              tensor_type {
-                               elem_type: FLOAT
+                               elem_type: 1
                                shape {
                                  dim {
                                     dim_value: 7
@@ -64,14 +64,14 @@ struct ConstMainFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxPars
 
 struct ConstValidFixture : ConstMainFixture
 {
-    ConstValidFixture() : ConstMainFixture("FLOAT") {
+    ConstValidFixture() : ConstMainFixture("1") {
         Setup();
     }
 };
 
 struct ConstInvalidFixture : ConstMainFixture
 {
-    ConstInvalidFixture() : ConstMainFixture("FLOAT16") { }
+    ConstInvalidFixture() : ConstMainFixture("10") { }
 };
 
 BOOST_FIXTURE_TEST_CASE(ValidConstTest, ConstValidFixture)
-- 
cgit v1.2.1