ArmNN
 21.02
Mean.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 (MeanNoAxisNoKeepDims, MeanNoAxisNoKeepDimsFixture)
 
 BOOST_FIXTURE_TEST_CASE (MeanWithAxis0NoKeepDims, MeanWithAxis0NoKeepDimsFixture)
 
 BOOST_FIXTURE_TEST_CASE (MeanWithAxis1NoKeepDims, MeanWithAxis1NoKeepDimsFixture)
 
 BOOST_FIXTURE_TEST_CASE (MeanWithAxis0KeepDims, MeanWithAxis0KeepDimsFixture)
 
 BOOST_FIXTURE_TEST_CASE (MeanWithAxis1KeepDims, MeanWithAxis1KeepDimsFixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/5]

BOOST_FIXTURE_TEST_CASE ( MeanNoAxisNoKeepDims  ,
MeanNoAxisNoKeepDimsFixture   
)

Definition at line 148 of file Mean.cpp.

149 {
150  RunTest<1>({ { "input", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f } } },
151  { { "output", { 1.5f } } });
152 }

◆ BOOST_FIXTURE_TEST_CASE() [2/5]

BOOST_FIXTURE_TEST_CASE ( MeanWithAxis0NoKeepDims  ,
MeanWithAxis0NoKeepDimsFixture   
)

Definition at line 154 of file Mean.cpp.

155 {
156  RunTest<1>({ { "input", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f } } },
157  { { "output", { 1.5f, 1.5f, 1.5f } } });
158 }

◆ BOOST_FIXTURE_TEST_CASE() [3/5]

BOOST_FIXTURE_TEST_CASE ( MeanWithAxis1NoKeepDims  ,
MeanWithAxis1NoKeepDimsFixture   
)

Definition at line 160 of file Mean.cpp.

161 {
162  RunTest<1>({ { "input", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f } } },
163  { { "output", { 1.f, 2.f } } });
164 }

◆ BOOST_FIXTURE_TEST_CASE() [4/5]

BOOST_FIXTURE_TEST_CASE ( MeanWithAxis0KeepDims  ,
MeanWithAxis0KeepDimsFixture   
)

Definition at line 166 of file Mean.cpp.

167 {
168  RunTest<2>({ { "input", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f } } },
169  { { "output", { 1.5f, 1.5f, 1.5f } } });
170 }

◆ BOOST_FIXTURE_TEST_CASE() [5/5]

BOOST_FIXTURE_TEST_CASE ( MeanWithAxis1KeepDims  ,
MeanWithAxis1KeepDimsFixture   
)

Definition at line 172 of file Mean.cpp.

References BOOST_AUTO_TEST_SUITE_END().

173 {
174  RunTest<2>({ { "input", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f } } },
175  { { "output", { 1.f, 2.f } } });
176 }