diff options
Diffstat (limited to 'src/armnnOnnxParser/test/Unsqueeze.cpp')
-rw-r--r-- | src/armnnOnnxParser/test/Unsqueeze.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/test/Unsqueeze.cpp b/src/armnnOnnxParser/test/Unsqueeze.cpp index 95a191e46b..7ba87bc680 100644 --- a/src/armnnOnnxParser/test/Unsqueeze.cpp +++ b/src/armnnOnnxParser/test/Unsqueeze.cpp @@ -77,6 +77,14 @@ struct UnsqueezeUnsortedAxesFixture : UnsqueezeFixture } }; +struct UnsqueezeScalarFixture : UnsqueezeFixture +{ + UnsqueezeScalarFixture() : UnsqueezeFixture({ 0 }, { }, { 1 }) + { + Setup(); + } +}; + TEST_CASE_FIXTURE(UnsqueezeSingleAxesFixture, "UnsqueezeSingleAxesTest") { RunTest<3, float>({{"Input", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f }}}, @@ -107,6 +115,12 @@ TEST_CASE_FIXTURE(UnsqueezeUnsortedAxesFixture, "UnsqueezeUnsortedAxesTest") 6.0f, 7.0f, 8.0f, 9.0f, 10.0f }}}); } +TEST_CASE_FIXTURE(UnsqueezeScalarFixture, "UnsqueezeScalarTest") +{ + RunTest<1, float>({{"Input", { 1.0f }}}, + {{"Output", { 1.0f }}}); +} + struct UnsqueezeInputAxesFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> { UnsqueezeInputAxesFixture() |