6 #include <boost/test/unit_test.hpp> 8 #include "../Deserializer.hpp" 17 explicit ArgMinMaxFixture(
const std::string& inputShape,
18 const std::string& outputShape,
19 const std::string& axis,
20 const std::string& argMinMaxFunction)
26 layer_type: "InputLayer", 30 layerName: "InputLayer", 38 dimensions: )" + inputShape + R"(, 40 quantizationScale: 0.0 49 layer_type: "ArgMinMaxLayer", 53 layerName: "ArgMinMaxLayer", 54 layerType: "ArgMinMax", 66 dimensions: )" + outputShape + R"(, 68 quantizationScale: 0.0 74 axis: )" + axis + R"(, 75 argMinMaxFunction: )" + argMinMaxFunction + R"( 80 layer_type: "OutputLayer", 85 layerName: "OutputLayer", 118 struct SimpleArgMinMaxFixture :
public ArgMinMaxFixture
120 SimpleArgMinMaxFixture() : ArgMinMaxFixture(
"[ 1, 1, 1, 5 ]",
128 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed64>(
130 {{
"InputLayer", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f}}},
131 {{
"OutputLayer",{ 3l }}});
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ArgMinMax, SimpleArgMinMaxFixture)
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
BOOST_AUTO_TEST_SUITE_END()