Gather.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (ParseGather, SimpleGatherFixture)
	BOOST_FIXTURE_TEST_CASE (ParseGatherUint8, GatherUint8Fixture)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/2]

BOOST_FIXTURE_TEST_CASE	(	ParseGather	,
		SimpleGatherFixture
	)

Definition at line 98 of file Gather.cpp.

{
    RunTest<2, armnn::DataType::Float32, armnn::DataType::Signed32, armnn::DataType::Float32>
        (0,
         {{ "inputTensor",  { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }}},
         {{ "indices", { 1, 3, 4 }}},
         {{ "outputTensor", { 3, 4, 7, 8, 9, 10 }}});
}

BOOST_FIXTURE_TEST_CASE() [2/2]

BOOST_FIXTURE_TEST_CASE	(	ParseGatherUint8	,
		GatherUint8Fixture
	)

Definition at line 112 of file Gather.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    RunTest<1, armnn::DataType::QAsymmU8, armnn::DataType::Signed32, armnn::DataType::QAsymmU8>
        (0,
         {{ "inputTensor",  { 1, 2, 3, 4, 5, 6, 7, 8 }}},
         {{ "indices", { 7, 6, 5 }}},
         {{ "outputTensor", { 8, 7, 6 }}});
}