ArmNN
 21.02
StridedSlice.cpp File Reference
#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>

Go to the source code of this file.

Functions

 BOOST_FIXTURE_TEST_CASE (StridedSlice4D, StridedSlice4DFixture)
 
 BOOST_FIXTURE_TEST_CASE (StridedSlice4DReverse, StridedSlice4DReverseFixture)
 
 BOOST_FIXTURE_TEST_CASE (StridedSliceSimpleStride, StridedSliceSimpleStrideFixture)
 
 BOOST_FIXTURE_TEST_CASE (StridedSliceSimpleRangeMask, StridedSliceSimpleRangeMaskFixture)
 

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/4]

BOOST_FIXTURE_TEST_CASE ( StridedSlice4D  ,
StridedSlice4DFixture   
)

Definition at line 118 of file StridedSlice.cpp.

119 {
120  RunTest<4, armnn::DataType::Float32>(
121  0,
122  {{"inputTensor", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
123 
124  3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
125 
126  5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f }}},
127 
128  {{"outputTensor", { 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f }}});
129 }

◆ BOOST_FIXTURE_TEST_CASE() [2/4]

BOOST_FIXTURE_TEST_CASE ( StridedSlice4DReverse  ,
StridedSlice4DReverseFixture   
)

Definition at line 150 of file StridedSlice.cpp.

151 {
152  RunTest<4, armnn::DataType::Float32>(
153  0,
154  {{"inputTensor", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
155 
156  3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
157 
158  5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f }}},
159 
160  {{"outputTensor", { 4.0f, 4.0f, 4.0f, 3.0f, 3.0f, 3.0f }}});
161 }

◆ BOOST_FIXTURE_TEST_CASE() [3/4]

BOOST_FIXTURE_TEST_CASE ( StridedSliceSimpleStride  ,
StridedSliceSimpleStrideFixture   
)

Definition at line 173 of file StridedSlice.cpp.

174 {
175  RunTest<4, armnn::DataType::Float32>(
176  0,
177  {{"inputTensor", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
178 
179  3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
180 
181  5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f }}},
182 
183  {{"outputTensor", { 1.0f, 1.0f,
184 
185  5.0f, 5.0f }}});
186 }

◆ BOOST_FIXTURE_TEST_CASE() [4/4]

BOOST_FIXTURE_TEST_CASE ( StridedSliceSimpleRangeMask  ,
StridedSliceSimpleRangeMaskFixture   
)

Definition at line 200 of file StridedSlice.cpp.

References BOOST_AUTO_TEST_SUITE_END().

201 {
202  RunTest<4, armnn::DataType::Float32>(
203  0,
204  {{"inputTensor", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
205 
206  3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
207 
208  5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f }}},
209 
210  {{"outputTensor", { 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
211 
212  3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
213 
214  5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f }}});
215 }