| 80 | } |
| 81 | |
| 82 | void 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 | |
| 115 | void MaxPool3dFP32Test(const std::vector<armnn::BackendId>& backends = {}) |
| 116 | { |