diff options
author | Pablo Tello <pablo.tello@arm.com> | 2019-12-11 13:04:34 +0000 |
---|---|---|
committer | SiCongLi <sicong.li@arm.com> | 2019-12-12 10:23:05 +0000 |
commit | e6b32020924144683e4f378f147bc9c1c2742e3c (patch) | |
tree | 2a9be2ba11426b650eeebc7af1bac1b0b102394f /tests/validation/CL/ReduceMean.cpp | |
parent | 0f76f9e08958bf195a610cef5e28cfb659a06c30 (diff) | |
download | ComputeLibrary-e6b32020924144683e4f378f147bc9c1c2742e3c.tar.gz |
COMPMID-2855: CLReduceMean throws error for invalid configs
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2452
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Change-Id: I7cda1b67aa6c3541fdd7781be12288c8fc36ffeb
Diffstat (limited to 'tests/validation/CL/ReduceMean.cpp')
-rw-r--r-- | tests/validation/CL/ReduceMean.cpp | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/tests/validation/CL/ReduceMean.cpp b/tests/validation/CL/ReduceMean.cpp index cfd4a2730c..1b7400bf53 100644 --- a/tests/validation/CL/ReduceMean.cpp +++ b/tests/validation/CL/ReduceMean.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -55,20 +55,26 @@ TEST_SUITE(ReduceMean) // *INDENT-OFF* // clang-format off -DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip( +DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(zip( framework::dataset::make("InputInfo", { TensorInfo(TensorShape(27U, 3U, 16U, 2U), 1, DataType::F32), // Invalid axis TensorInfo(TensorShape(27U, 3U, 16U, 2U), 1, DataType::F32), // Invalid output shape - TensorInfo(TensorShape(32U, 16U, 16U, 2U), 1, DataType::F32) + TensorInfo(TensorShape(32U, 16U, 16U, 2U), 1, DataType::F32),// OK + TensorInfo(TensorShape{228U, 19U, 2U, 2U}, 1, DataType::F32),// OK + TensorInfo(TensorShape{228U, 19U, 2U, 1U}, 1, DataType::F32) // Cannot support axis 3 not valid }), framework::dataset::make("OutputInfo", { TensorInfo(TensorShape(27U, 3U, 1U, 2U), 1, DataType::F32), TensorInfo(TensorShape(27U, 3U, 1U, 2U), 1, DataType::F32), - TensorInfo(TensorShape(32U, 16U, 1U, 2U), 1, DataType::F32) + TensorInfo(TensorShape(32U, 16U, 1U, 2U), 1, DataType::F32), + TensorInfo(TensorShape(19U), 1, DataType::F32), + TensorInfo(TensorShape(19U), 1, DataType::F32) + })), - framework::dataset::make("Axis", { Coordinates(4), Coordinates(0,2), Coordinates(2) })), - framework::dataset::make("Expected", { false, false, true })), - input_info, output_info, axis, expected) + framework::dataset::make("Axis", { Coordinates(4), Coordinates(0,2), Coordinates(2), Coordinates(3,2,0), Coordinates(3,2,0) })), + framework::dataset::make("Keep", { true, true, true, false, false })), + framework::dataset::make("Expected", { false, false, true, true, false })), + input_info, output_info, axis, keep, expected) { - const Status status = CLReduceMean::validate(&input_info.clone()->set_is_resizable(false), axis, true, &output_info.clone()->set_is_resizable(false)); + const Status status = CLReduceMean::validate(&input_info.clone()->set_is_resizable(false), axis, keep, &output_info.clone()->set_is_resizable(false)); ARM_COMPUTE_EXPECT(bool(status) == expected, framework::LogLevel::ERRORS); } // clang-format on |