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

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (ParseFusedBatchNormNhwc, FusedBatchNormNhwcFixture)
 
 BOOST_FIXTURE_TEST_CASE (ParseFusedBatchNormNchw, FusedBatchNormNchwFixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/2]

BOOST_FIXTURE_TEST_CASE ( ParseFusedBatchNormNhwc  ,
FusedBatchNormNhwcFixture   
)

Definition at line 192 of file FusedBatchNorm.cpp.

193 {
194  RunTest<4>({ 1, 2, 3, 4, 5, 6, 7, 8, 9 }, // Input data.
195  { -2.8277204f, -2.12079024f, -1.4138602f,
196  -0.7069301f, 0.0f, 0.7069301f,
197  1.4138602f, 2.12079024f, 2.8277204f }); // Expected output data.
198 }

◆ BOOST_FIXTURE_TEST_CASE() [2/2]

BOOST_FIXTURE_TEST_CASE ( ParseFusedBatchNormNchw  ,
FusedBatchNormNchwFixture   
)

Definition at line 204 of file FusedBatchNorm.cpp.

References BOOST_AUTO_TEST_SUITE_END().

205 {
206  RunTest<4>({ 1, 2, 3, 4, 5, 6, 7, 8, 9 }, // Input data.
207  { -2.8277204f, -2.12079024f, -1.4138602f,
208  -0.7069301f, 0.0f, 0.7069301f,
209  1.4138602f, 2.12079024f, 2.8277204f }); // Expected output data.
210 }