ArmNN
 21.02
FullyConnected.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "armnnTfParser/ITfParser.hpp"
#include "ParserPrototxtFixture.hpp"
#include "Runtime.hpp"
#include "Network.hpp"
#include "Graph.hpp"

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (FullyConnected, FullyConnectedFixture)
 
 BOOST_FIXTURE_TEST_CASE (MatMulUsedInTwoFc, MatMulUsedInTwoFcFixture)
 
 BOOST_FIXTURE_TEST_CASE (MatMulUsedInTwoFcStaggered, MatMulUsedInTwoFcStaggeredFixture)
 
 BOOST_FIXTURE_TEST_CASE (MatMul, MatMulFixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/4]

BOOST_FIXTURE_TEST_CASE ( FullyConnected  ,
FullyConnectedFixture   
)

Definition at line 142 of file FullyConnected.cpp.

143 {
144  RunTest<1>({ 3 }, { 7 });
145 }

◆ BOOST_FIXTURE_TEST_CASE() [2/4]

BOOST_FIXTURE_TEST_CASE ( MatMulUsedInTwoFc  ,
MatMulUsedInTwoFcFixture   
)

Definition at line 324 of file FullyConnected.cpp.

325 {
326  RunTest<1>({ 3 }, { 32 });
327  // Ideally we would check here that the armnn network has 5 layers:
328  // Input, 2 x FullyConnected (biased), Add and Output.
329  // This would make sure the parser hasn't incorrectly added some unconnected layers corresponding to the MatMul.
330 }

◆ BOOST_FIXTURE_TEST_CASE() [3/4]

BOOST_FIXTURE_TEST_CASE ( MatMulUsedInTwoFcStaggered  ,
MatMulUsedInTwoFcStaggeredFixture   
)

Definition at line 479 of file FullyConnected.cpp.

480 {
481  RunTest<1>({ 2 }, { 75 });
482  // Ideally we would check here that the armnn network has 5 layers:
483  // Input, FullyConnected (biased), FullyConnected (non biased), Add and Output.
484 }

◆ BOOST_FIXTURE_TEST_CASE() [4/4]

BOOST_FIXTURE_TEST_CASE ( MatMul  ,
MatMulFixture   
)

Definition at line 574 of file FullyConnected.cpp.

References BOOST_AUTO_TEST_SUITE_END().

575 {
576  RunTest<1>({ 2 }, { 34 });
577 }