diff options
author | Lev Proleev <levp@google.com> | 2020-03-26 13:40:05 +0000 |
---|---|---|
committer | Android (Google) Code Review <android-gerrit@google.com> | 2020-03-26 13:40:05 +0000 |
commit | a3493bc02b986a542bb55f8a3585b3e44bb4ad1d (patch) | |
tree | b7be8f3e830363400657607b15e1c5695569aa8d /nn/runtime/test/TestGenerated.cpp | |
parent | b1fdc0a0373b62bcadd34d7045a526f27c395a6a (diff) | |
parent | 44f40827d8e41055f18c1a313701785da74f6c15 (diff) | |
download | ml-a3493bc02b986a542bb55f8a3585b3e44bb4ad1d.tar.gz |
Merge "Add rank check to concatenation op" into rvc-dev
Diffstat (limited to 'nn/runtime/test/TestGenerated.cpp')
-rw-r--r-- | nn/runtime/test/TestGenerated.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nn/runtime/test/TestGenerated.cpp b/nn/runtime/test/TestGenerated.cpp index 0e90113d6..60f866cab 100644 --- a/nn/runtime/test/TestGenerated.cpp +++ b/nn/runtime/test/TestGenerated.cpp @@ -351,6 +351,9 @@ void GeneratedTests::execute(const TestModel& testModel) { NNTRACE_APP(NNTRACE_PHASE_OVERALL, "execute"); GeneratedModel model; createModel(testModel, mTestDynamicOutputShape, &model); + if (testModel.expectFailure && !model.isValid()) { + return; + } ASSERT_EQ(model.finish(), Result::NO_ERROR); ASSERT_TRUE(model.isValid()); auto executeInternal = [&testModel, &model, this]() { |