| 47 | } |
| 48 | |
| 49 | void 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 | |
| 82 | void MaxPool3dFP32H1Test(const std::vector<armnn::BackendId>& backends = {}) |
| 83 | { |