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

Function AveragePool2dFP32PaddingSameTest

delegate/test/Pooling2dTest.cpp:406–430  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

404}
405
406void AveragePool2dFP32PaddingSameTest(const std::vector<armnn::BackendId>& backends = {})
407{
408 // Set input data
409 std::vector<int32_t> inputShape { 1, 3, 4, 1 };
410 std::vector<int32_t> outputShape { 1, 2, 2, 1 };
411
412 std::vector<float> inputValues = { -5.0f, 8.0f, -10.0f, 7.0f,
413 8.0f, 12.0f, -15.0f, 2.0f,
414 3.0f, -4.0f, -1.0f, -11.0f };
415
416 std::vector<float> expectedOutputValues = { 5.75f, -4.0f, -0.5f, -6.0f };
417
418 Pooling2dTest<float>(tflite::BuiltinOperator_AVERAGE_POOL_2D,
419 ::tflite::TensorType_FLOAT32,
420 inputShape,
421 outputShape,
422 inputValues,
423 expectedOutputValues,
424 backends,
425 ::tflite::Padding_SAME,
426 2,
427 2,
428 2,
429 2);
430}
431
432void AveragePool2dInt8PaddingSameTest(const std::vector<armnn::BackendId>& backends = {})
433{

Callers 1

Pooling2dTest.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected