aboutsummaryrefslogtreecommitdiff
path: root/src/armnnOnnxParser/test/Const.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnOnnxParser/test/Const.cpp')
-rw-r--r--src/armnnOnnxParser/test/Const.cpp8
1 files changed, 4 insertions, 4 deletions
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)