MCPcopy Create free account
hub / github.com/ARM-software/armnn / AveragePool2dInt8ReluTest

Function AveragePool2dInt8ReluTest

delegate/test/Pooling2dTest.cpp:488–515  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

486}
487
488void 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
517void AveragePool2dFP32Relu6Test(const std::vector<armnn::BackendId>& backends = {})
518{

Callers 1

Pooling2dTest.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected