30 #include "tests/datasets/Pooling3dLayerDataset.h"
31 #include "tests/datasets/PoolingTypesDataset.h"
32 #include "tests/datasets/ShapeDatasets.h"
37 #include "tests/validation/fixtures/Pooling3dLayerFixture.h"
49 framework::dataset::make(
"Stride", { Size3D(1, 1, 1), Size3D(2, 1, 1), Size3D(1, 2, 1), Size3D(2, 2, 1) })),
55 framework::dataset::make(
"Padding", { Padding3D(0, 0, 0), Padding3D(1, 1, 1), Padding3D(1, 0, 0) })),
60 framework::dataset::make(
"Stride", { Size3D(1, 1, 1), Size3D(2, 1, 1), Size3D(1, 2, 1), Size3D(2, 2, 1) })),
61 framework::dataset::make(
"Padding", { Padding3D(0, 0, 0), Padding3D(1, 1, 1), Padding3D(1, 0, 0) })),
70 using ShapeDataset = framework::dataset::ContainerDataset<std::vector<TensorShape>>;
73 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
76 constexpr AbsoluteTolerance<uint8_t> tolerance_qasymm8(1);
77 constexpr AbsoluteTolerance<int8_t> tolerance_qasymm8_s(1);
82 QuantizationInfo(.2f, 10),
83 QuantizationInfo(.1f, 5),
84 QuantizationInfo(2.f, 3)
87 const auto qasymm8_signed_in_qinfo_dataset =
framework::dataset::make(
"InputQuantInfo", { QuantizationInfo(.2f, -10) });
90 QuantizationInfo(.2f, -10),
91 QuantizationInfo(.1f, -5),
92 QuantizationInfo(2.f, -3)
147 Pooling3dLayerInfo(
PoolingType::AVG, 3,
Size3D(2U, 2U, 2U),
Padding3D(2,1,2,2,1,2),
false,
false,
DimensionRoundingType::CEIL),
149 framework::dataset::make(
"Expected", {
false,
false,
false,
false,
true,
false,
false,
false,
false, true ,
false,
true,
false,
false,
false})),
158 template <
typename T>
161 template <
typename T>
164 template <
typename T>
242 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
267 TensorShape(4U, 27U, 13U, 4U, 2U)
284 TensorShape(27U, 13U, 4U, 4U, 2U)
296 TensorShape(4U, 79U, 37U, 11U, 2U)
317 template <typename T>
318 using NEPooling3dLayerQuantizedFixture = Pooling3dLayerValidationQuantizedFixture<Tensor, Accessor, NEPooling3dLayer, T>;
322 combine(Pooling3dLayerDatasetQASYMM8Small,
324 qasymm8_in_qinfo_dataset),
325 qasymm8_out_qinfo_dataset))
332 combine(Pooling3dLayerDatasetQASYMM8Large,
334 qasymm8_in_qinfo_dataset),
335 qasymm8_out_qinfo_dataset))
346 combine(Pooling3dLayerDatasetQASYMM8Small,
348 qasymm8_signed_in_qinfo_dataset),
349 qasymm8_signed_out_qinfo_dataset))