ArmNN
 21.02
Gather.cpp File Reference
#include "armnnTfParser/ITfParser.hpp"
#include "ParserPrototxtFixture.hpp"
#include <PrototxtConversions.hpp>
#include <boost/test/unit_test.hpp>

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (ParseGather1DParams1DIndices, GatherFixture1DParams1DIndices)
 
 BOOST_FIXTURE_TEST_CASE (ParseGather1DParamsMultiDimIndices, GatherFixture1DParamsMultiDimIndices)
 
 BOOST_FIXTURE_TEST_CASE (ParseGatherMultiDimParamMultiDimIndices, GatherFixtureMultiDimParamMultiDimIndices)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/3]

BOOST_FIXTURE_TEST_CASE ( ParseGather1DParams1DIndices  ,
GatherFixture1DParams1DIndices   
)

Definition at line 163 of file Gather.cpp.

164 {
165  RunTest<4>({ { "input0", { 1, 2, 3, 4 } } },
166 
167  { { "output", { 1, 3, 2, 4 } } });
168 }

◆ BOOST_FIXTURE_TEST_CASE() [2/3]

BOOST_FIXTURE_TEST_CASE ( ParseGather1DParamsMultiDimIndices  ,
GatherFixture1DParamsMultiDimIndices   
)

Definition at line 170 of file Gather.cpp.

171 {
172  RunTest<4>({ { "input0", { 1, 2, 3, 4 } } },
173 
174  { { "output", { 1, 2, 2, 4 } } });
175 }

◆ BOOST_FIXTURE_TEST_CASE() [3/3]

BOOST_FIXTURE_TEST_CASE ( ParseGatherMultiDimParamMultiDimIndices  ,
GatherFixtureMultiDimParamMultiDimIndices   
)

Definition at line 177 of file Gather.cpp.

References BOOST_AUTO_TEST_SUITE_END().

178 {
179  RunTest<4>({ { "input0", { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 } } },
180 
181  { { "output", { 3, 4, 7, 8, 1, 2, 5, 6} } });
182 }