diff options
Diffstat (limited to 'tests/validation_new/Validation.h')
-rw-r--r-- | tests/validation_new/Validation.h | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/validation_new/Validation.h b/tests/validation_new/Validation.h index 0766636c78..2892fb3e42 100644 --- a/tests/validation_new/Validation.h +++ b/tests/validation_new/Validation.h @@ -283,8 +283,13 @@ void validate(const IAccessor &tensor, const SimpleTensor<T> &reference, const V int64_t num_elements = 0; ARM_COMPUTE_EXPECT_EQUAL(tensor.element_size(), reference.element_size(), framework::LogLevel::ERRORS); - ARM_COMPUTE_EXPECT_EQUAL(tensor.format(), reference.format(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT_EQUAL(tensor.data_type(), reference.data_type(), framework::LogLevel::ERRORS); + + if(reference.format() != Format::UNKNOWN) + { + ARM_COMPUTE_EXPECT_EQUAL(tensor.format(), reference.format(), framework::LogLevel::ERRORS); + } + ARM_COMPUTE_EXPECT_EQUAL(tensor.num_channels(), reference.num_channels(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT(compare_dimensions(tensor.shape(), reference.shape()), framework::LogLevel::ERRORS); |