ArmNN
 21.02
TestConvolution.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "armnnCaffeParser/ICaffeParser.hpp"
#include "ParserPrototxtFixture.hpp"
#include <sstream>
#include <initializer_list>

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (SimpleConvolution, SimpleConvolutionFixture)
 
 BOOST_FIXTURE_TEST_CASE (GroupConvolution, GroupConvolutionFixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/2]

BOOST_FIXTURE_TEST_CASE ( SimpleConvolution  ,
SimpleConvolutionFixture   
)

Definition at line 97 of file TestConvolution.cpp.

98 {
99  RunTest<4>({ 1, 3, 5, 7 }, { 16 });
100 }

◆ BOOST_FIXTURE_TEST_CASE() [2/2]

BOOST_FIXTURE_TEST_CASE ( GroupConvolution  ,
GroupConvolutionFixture   
)

Definition at line 118 of file TestConvolution.cpp.

References BOOST_AUTO_TEST_SUITE_END().

119 {
120  RunTest<4>(
121  {
122  1, 2, 3, 4, // input channel #0
123  5, 6, 7, 8, // input channel #1
124  },
125  {
126  10, // convolution result for channel #0 applying filter #0
127  52 // same for channel #1 and filter #1
128  }
129  );
130 }