| 486 | } |
| 487 | |
| 488 | void AveragePool2dInt8ReluTest(const std::vector<armnn::BackendId>& backends = {}) |
| 489 | { |
| 490 | // Set input data |
| 491 | std::vector<int32_t> inputShape { 1, 3, 4, 1 }; |
| 492 | std::vector<int32_t> outputShape { 1, 2, 3, 1 }; |
| 493 | |
| 494 | std::vector<int8_t> inputValues = { -5, 8, -10, 7, |
| 495 | -8, 12, -15, 2, |
| 496 | 3, -4, -1, 11 }; |
| 497 | |
| 498 | std::vector<int8_t> expectedOutputValues = { 2, 1, 1, 1, 1, 1 }; |
| 499 | |
| 500 | Pooling2dTest<int8_t>(tflite::BuiltinOperator_AVERAGE_POOL_2D, |
| 501 | ::tflite::TensorType_INT8, |
| 502 | inputShape, |
| 503 | outputShape, |
| 504 | inputValues, |
| 505 | expectedOutputValues, |
| 506 | backends, |
| 507 | ::tflite::Padding_VALID, |
| 508 | 1, |
| 509 | 1, |
| 510 | 2, |
| 511 | 2, |
| 512 | ::tflite::ActivationFunctionType_RELU, |
| 513 | 2.5f, |
| 514 | 1); |
| 515 | } |
| 516 | |
| 517 | void AveragePool2dFP32Relu6Test(const std::vector<armnn::BackendId>& backends = {}) |
| 518 | { |