diff options
author | Usama Arif <usama.arif@arm.com> | 2019-03-21 11:49:25 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-03-21 13:52:05 +0000 |
commit | c63c3703db0bf351fc8d17fc1c0d3721abb3cc80 (patch) | |
tree | f441a18ef7b9ad71d3d4f9e872bece293469154a /tests/validation/NEON/WidthConcatenateLayer.cpp | |
parent | 89890c630c438debe9661eeb444d81d6e2b3e1a5 (diff) | |
download | ComputeLibrary-c63c3703db0bf351fc8d17fc1c0d3721abb3cc80.tar.gz |
COMPMID-2074: Fix WidthConcatenate and HeightConcatenate validate tests
Change-Id: I14132883047d7df478e8e30917cb8bbaadd93be7
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/883
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/NEON/WidthConcatenateLayer.cpp')
-rw-r--r-- | tests/validation/NEON/WidthConcatenateLayer.cpp | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/tests/validation/NEON/WidthConcatenateLayer.cpp b/tests/validation/NEON/WidthConcatenateLayer.cpp index dba14ebb35..20df3f4d7d 100644 --- a/tests/validation/NEON/WidthConcatenateLayer.cpp +++ b/tests/validation/NEON/WidthConcatenateLayer.cpp @@ -47,19 +47,22 @@ DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip( framework::dataset::make("InputInfo1", { TensorInfo(TensorShape(23U, 27U, 5U), 1, DataType::F32), // Mismatching data type input/output TensorInfo(TensorShape(23U, 27U, 5U), 1, DataType::F32), // Mismatching y dimension TensorInfo(TensorShape(23U, 27U, 5U), 1, DataType::F32), // Mismatching total width - TensorInfo(TensorShape(16U, 27U, 5U), 1, DataType::F32) + TensorInfo(TensorShape(16U, 27U, 5U), 1, DataType::F32), + TensorInfo(TensorShape(21U, 35U, 5U), 1, DataType::F32) }), framework::dataset::make("InputInfo2", { TensorInfo(TensorShape(24U, 27U, 4U), 1, DataType::F32), TensorInfo(TensorShape(52U, 27U, 5U), 1, DataType::F32), TensorInfo(TensorShape(52U, 27U, 5U), 1, DataType::F32), - TensorInfo(TensorShape(16U, 27U, 5U), 1, DataType::F32) + TensorInfo(TensorShape(16U, 27U, 5U), 1, DataType::F32), + TensorInfo(TensorShape(10U, 35U, 5U), 1, DataType::F32) })), framework::dataset::make("OutputInfo", { TensorInfo(TensorShape(47U, 27U, 5U), 1, DataType::F16), TensorInfo(TensorShape(75U, 12U, 5U), 1, DataType::F32), TensorInfo(TensorShape(11U, 27U, 5U), 1, DataType::F32), - TensorInfo(TensorShape(32U, 27U, 5U), 1, DataType::F32) + TensorInfo(TensorShape(32U, 27U, 5U), 1, DataType::F32), + TensorInfo(TensorShape(31U, 35U, 5U), 1, DataType::F32) })), - framework::dataset::make("Expected", { false, false, false, true })), + framework::dataset::make("Expected", { false, false, false, true, true })), input_info1, input_info2, output_info,expected) { std::vector<TensorInfo> inputs_vector_info; @@ -73,7 +76,7 @@ DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip( } bool is_valid = bool(NEConcatenateLayer::validate(inputs_vector_info_raw, - &output_info.clone()->set_is_resizable(false),DataLayoutDimension::WIDTH)); + &output_info.clone()->set_is_resizable(true),DataLayoutDimension::WIDTH)); ARM_COMPUTE_EXPECT(is_valid == expected, framework::LogLevel::ERRORS); } // clang-format on |