Function TEST_CASE

tests/validation/NEON/ReduceMean.cpp:82–117

TEST_SUITE(ReduceMean)

TEST_CASE(ProperRoundingPolicyXReduction, framework::DatasetMode::ALL)
{
    // We do not need to stress vector and leftover loops diffrently
    // because the rounding is done scalarly at the end. Accumulation
    // is done over integer types.
    constexpr int x_len = 2;

    const auto input_shape  = TensorShape(x_len);
    const auto output_shape = TensorShape(1);
    const bool keep_dims    = true;
    const auto axis         = Coordinates(0);
    const auto input_qinfo  = QuantizationInfo(2 / 255.f, 0);
    const auto output_qinfo = QuantizationInfo(6 / 255.f, -1);
    const auto dtype        = DataType::QASYMM8_SIGNED;

    Tensor input  = create_tensor<Tensor>(input_shape, dtype, 1, input_qinfo);
    Tensor output = create_tensor<Tensor>(output_shape, dtype, 1, output_qinfo);

    NEReduceMean reduce_mean;
    reduce_mean.configure(&input, axis, keep_dims, &output);

    input.allocator()->allocate();
    output.allocator()->allocate();

    std::vector<int8_t> values{50, 26};
    library->fill_static_values(Accessor(input), values);

    std::vector<int8_t>  expected{12};
    SimpleTensor<int8_t> ref{output_shape, dtype, 1, input_qinfo};
    library->fill_static_values(ref, expected);

    reduce_mean.run();

    // The tolerance should be 0 because this test stresses the rounding behavior of the operator
    validate(Accessor(output), ref, zero_tolerance_s8);
}

#ifdef __aarch64__
// Due to the lack of instructions in a32, the rounding operation is less