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

Function MaxPool3dFP32H1Test

delegate/test/Pooling3dTest.cpp:82–113  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

80}
81
82void MaxPool3dFP32H1Test(const std::vector<armnn::BackendId>& backends = {})
83{
84 // Set input data and expected output data
85 std::vector<int32_t> inputShape = { 1, 2, 3, 4, 1 };
86 std::vector<int32_t> outputShape = { 1, 1, 3, 3, 1 };
87
88 std::vector<float> inputValues = { 1, 2, 3, 4, 5, 6,
89 1, 2, 3, 4, 5, 6,
90 1, 2, 3, 4, 5, 6,
91 1, 2, 3, 4, 5, 6 };
92 std::vector<float> expectedOutputValues = { 2, 3 };
93
94 // poolType string required to create the correct pooling operator
95 // Padding type required to create the padding in custom options
96 std::string poolType = "kMax";
97 TfLitePadding padding = kTfLitePaddingValid;
98
99 Pooling3dTest<float>(poolType,
100 ::tflite::TensorType_FLOAT32,
101 inputShape,
102 outputShape,
103 inputValues,
104 expectedOutputValues,
105 backends,
106 padding,
107 1,
108 1,
109 1,
110 2,
111 1,
112 2);
113}
114
115void MaxPool3dFP32Test(const std::vector<armnn::BackendId>& backends = {})
116{

Callers 1

Pooling3dTest.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected