diff options
Diffstat (limited to 'test/Convolution2D.hpp')
-rw-r--r-- | test/Convolution2D.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/Convolution2D.hpp b/test/Convolution2D.hpp index 38216f10..c471fbd7 100644 --- a/test/Convolution2D.hpp +++ b/test/Convolution2D.hpp @@ -104,7 +104,10 @@ void PaddingTestImpl(android::nn::PaddingScheme paddingScheme, bool fp16Enabled float* outdata = reinterpret_cast<float*>(static_cast<void*>(outMemory->getPointer())); // run the execution - Execute(preparedModel, request); + if (preparedModel.get() != nullptr) + { + Execute(preparedModel, request); + } // check the result switch (paddingScheme) |