| 629 | } |
| 630 | |
| 631 | void AveragePool2dInt16PaddingSameTest(const std::vector<armnn::BackendId>& backends = {}) |
| 632 | { |
| 633 | // Set input data |
| 634 | std::vector<int32_t> inputShape { 1, 3, 4, 1 }; |
| 635 | std::vector<int32_t> outputShape { 1, 2, 2, 1 }; |
| 636 | |
| 637 | std::vector<int16_t > inputValues = { -5, 8, -10, 7, |
| 638 | 8, 12, -15, 2, |
| 639 | 3, -4, -1, -11 }; |
| 640 | |
| 641 | std::vector<int16_t> expectedOutputValues = { 6, -4, -1, -6 }; |
| 642 | |
| 643 | Pooling2dTest<int16_t>(tflite::BuiltinOperator_AVERAGE_POOL_2D, |
| 644 | ::tflite::TensorType_INT16, |
| 645 | inputShape, |
| 646 | outputShape, |
| 647 | inputValues, |
| 648 | expectedOutputValues, |
| 649 | backends, |
| 650 | ::tflite::Padding_SAME, |
| 651 | 2, |
| 652 | 2, |
| 653 | 2, |
| 654 | 2, |
| 655 | tflite::ActivationFunctionType_NONE, |
| 656 | 2.5f, |
| 657 | 0); |
| 658 | } |
| 659 | |
| 660 | void AveragePool2dInt16ReluTest(const std::vector<armnn::BackendId>& backends = {}) |
| 661 | { |