| 571 | } |
| 572 | |
| 573 | void AveragePool2dUint8PaddingSameTest(const std::vector<armnn::BackendId>& backends = {}) |
| 574 | { |
| 575 | // Set input data |
| 576 | std::vector<int32_t> inputShape { 1, 3, 4, 1 }; |
| 577 | std::vector<int32_t> outputShape { 1, 2, 2, 1 }; |
| 578 | |
| 579 | std::vector<uint8_t> inputValues = { 5, 8, 10, 7, |
| 580 | 8, 12, 15, 2, |
| 581 | 3, 4, 1, 11 }; |
| 582 | |
| 583 | std::vector<uint8_t> expectedOutputValues = { 8, 9, 4, 6 }; |
| 584 | |
| 585 | Pooling2dTest<uint8_t>(tflite::BuiltinOperator_AVERAGE_POOL_2D, |
| 586 | ::tflite::TensorType_UINT8, |
| 587 | inputShape, |
| 588 | outputShape, |
| 589 | inputValues, |
| 590 | expectedOutputValues, |
| 591 | backends, |
| 592 | ::tflite::Padding_SAME, |
| 593 | 2, |
| 594 | 2, |
| 595 | 2, |
| 596 | 2, |
| 597 | tflite::ActivationFunctionType_NONE, |
| 598 | 2.5f, |
| 599 | 1); |
| 600 | } |
| 601 | |
| 602 | void AveragePool2dUint8ReluTest(const std::vector<armnn::BackendId>& backends = {}) |
| 603 | { |