diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-09-15 17:30:37 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-09-16 09:08:39 +0000 |
commit | f106ab745a12a5c773a9c315dcddef0c8bf11225 (patch) | |
tree | d0d3f2ca4e084c044e50fc1950f2fe933e309992 /src/armnnOnnxParser/test/Clip.cpp | |
parent | 7ba84d6881685d6ebfedc597a9af98b16fa42d51 (diff) | |
download | armnn-f106ab745a12a5c773a9c315dcddef0c8bf11225.tar.gz |
Add support for Clip with attribute on ONNX parser
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I1bae42dade7eabf3da09252066e912e803a8ea32
Diffstat (limited to 'src/armnnOnnxParser/test/Clip.cpp')
-rw-r--r-- | src/armnnOnnxParser/test/Clip.cpp | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/test/Clip.cpp b/src/armnnOnnxParser/test/Clip.cpp index b0447bcad5..2b43574d6c 100644 --- a/src/armnnOnnxParser/test/Clip.cpp +++ b/src/armnnOnnxParser/test/Clip.cpp @@ -62,6 +62,68 @@ struct ClipMainFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParse } }; +struct ClipAttributeFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> +{ + ClipAttributeFixture(std::string min, std::string max) + { + m_Prototext = R"( + ir_version: 3 + producer_name: "CNTK" + producer_version: "2.5.1" + domain: "ai.cntk" + model_version: 1 + graph { + name: "CNTKGraph" + input { + name: "Input" + type { + tensor_type { + elem_type: 1 + shape { + dim { + dim_value: 5 + } + } + } + } + } + node { + input: "Input" + output: "Output" + name: "ActivationLayer" + op_type: "Clip" + attribute { + name: "min" + f: )" + min + R"( + type: FLOAT + } + attribute { + name: "max" + f: )" + max + R"( + type: FLOAT + } + } + output { + name: "Output" + type { + tensor_type { + elem_type: 1 + shape { + dim { + dim_value: 5 + } + } + } + } + } + } + opset_import { + version: 7 + })"; + Setup(); + } +}; + struct ClipFixture : ClipMainFixture { ClipFixture() : ClipMainFixture("2", "3.5") {} @@ -108,4 +170,15 @@ TEST_CASE_FIXTURE(ClipNoInputFixture, "ValidNoInputClipTest") std::numeric_limits<float>::max()}}}); } +struct ClipMinMaxAttributeFixture : ClipAttributeFixture +{ + ClipMinMaxAttributeFixture() : ClipAttributeFixture("2", "3.5") {} +}; + +TEST_CASE_FIXTURE(ClipMinMaxAttributeFixture, "ValidClipAttributeTest") +{ + RunTest<1>({{ "Input", { -1.5f, 1.25f, 3.5f, 8.0, 2.5}}}, + {{ "Output", { 2.0f, 2.0f, 3.5f, 3.5, 2.5}}}); +} + } |