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
diff --git a/src/armnnOnnxParser/test/Pooling.cpp b/src/armnnOnnxParser/test/Pooling.cpp
index cdb9b4f..85bf914 100644
--- a/src/armnnOnnxParser/test/Pooling.cpp
+++ b/src/armnnOnnxParser/test/Pooling.cpp
@@ -73,7 +73,7 @@
name: "Output"
type {
tensor_type {
- elem_type: FLOAT
+ elem_type: 1
shape {
dim {
dim_value: 1
@@ -100,14 +100,14 @@
struct MaxPoolValidFixture : PoolingMainFixture
{
- MaxPoolValidFixture() : PoolingMainFixture("FLOAT", "\"MaxPool\"") {
+ MaxPoolValidFixture() : PoolingMainFixture("1", "\"MaxPool\"") {
Setup();
}
};
struct MaxPoolInvalidFixture : PoolingMainFixture
{
- MaxPoolInvalidFixture() : PoolingMainFixture("FLOAT16", "\"MaxPool\"") { }
+ MaxPoolInvalidFixture() : PoolingMainFixture("10", "\"MaxPool\"") { }
};
BOOST_FIXTURE_TEST_CASE(ValidMaxPoolTest, MaxPoolValidFixture)
@@ -117,7 +117,7 @@
struct AvgPoolValidFixture : PoolingMainFixture
{
- AvgPoolValidFixture() : PoolingMainFixture("FLOAT", "\"AveragePool\"") {
+ AvgPoolValidFixture() : PoolingMainFixture("1", "\"AveragePool\"") {
Setup();
}
};
@@ -138,7 +138,7 @@
name: "Input"
type {
tensor_type {
- elem_type: FLOAT
+ elem_type: 1
shape {
dim {
dim_value: 1
@@ -191,7 +191,7 @@
name: "Output"
type {
tensor_type {
- elem_type: FLOAT
+ elem_type: 1
shape {
dim {
dim_value: 1
@@ -243,7 +243,7 @@
name: "Input"
type {
tensor_type {
- elem_type: FLOAT
+ elem_type: 1
shape {
dim {
dim_value: 1
@@ -271,7 +271,7 @@
name: "Output"
type {
tensor_type {
- elem_type: FLOAT
+ elem_type: 1
shape {
dim {
dim_value: 1