7 #include "../Deserializer.hpp" 16 explicit ArgMinMaxFixture(
const std::string& inputShape,
17 const std::string& outputShape,
18 const std::string& axis,
19 const std::string& argMinMaxFunction)
25 layer_type: "InputLayer", 29 layerName: "InputLayer", 37 dimensions: )" + inputShape + R"(, 39 quantizationScale: 0.0 48 layer_type: "ArgMinMaxLayer", 52 layerName: "ArgMinMaxLayer", 53 layerType: "ArgMinMax", 65 dimensions: )" + outputShape + R"(, 67 quantizationScale: 0.0 73 axis: )" + axis + R"(, 74 argMinMaxFunction: )" + argMinMaxFunction + R"( 79 layer_type: "OutputLayer", 84 layerName: "OutputLayer", 117 struct SimpleArgMinMaxFixture :
public ArgMinMaxFixture
119 SimpleArgMinMaxFixture() : ArgMinMaxFixture(
"[ 1, 1, 1, 5 ]",
127 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed64>(
129 {{
"InputLayer", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f}}},
130 {{
"OutputLayer",{ 3l }}});
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("DeserializeParser_ArgMinMax")