MCPcopy Create free account
hub / github.com/ARM-software/armnn / MaxPool3dFP32PaddingSameTest

Function MaxPool3dFP32PaddingSameTest

delegate/test/Pooling3dTest.cpp:49–80  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

47}
48
49void MaxPool3dFP32PaddingSameTest(const std::vector<armnn::BackendId>& backends = {})
50{
51 // Set input data and expected output data
52 std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 };
53 std::vector<int32_t> outputShape = { 1, 2, 3, 4, 1 };
54
55 std::vector<float> inputValues = { 1, 2, 3, 4, 5, 6,
56 1, 2, 3, 4, 5, 6,
57 1, 2, 3, 4, 5, 6,
58 1, 2, 3, 4, 5, 6 };
59 std::vector<float> expectedOutputValues = { 6, 6, 4, 4, 6, 6, 6, 6, 4, 5, 6, 6, 6, 6, 4, 4 };
60
61 // poolType string required to create the correct pooling operator
62 // Padding type required to create the padding in custom options
63 std::string poolType = "kMax";
64 TfLitePadding padding = kTfLitePaddingSame;
65
66 Pooling3dTest<float>(poolType,
67 ::tflite::TensorType_FLOAT32,
68 inputShape,
69 outputShape,
70 inputValues,
71 expectedOutputValues,
72 backends,
73 padding,
74 1,
75 1,
76 1,
77 2,
78 2,
79 2);
80}
81
82void MaxPool3dFP32H1Test(const std::vector<armnn::BackendId>& backends = {})
83{

Callers 1

Pooling3dTest.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected