diff options
author | Miao Wang <miaowang@google.com> | 2020-02-25 16:54:53 -0800 |
---|---|---|
committer | Slava Shklyaev <slavash@google.com> | 2020-03-13 13:45:47 +0000 |
commit | f65effce2817e390784aa7f2d0c5cf3334694db8 (patch) | |
tree | deb45379d524a7ec1bc044aeb53eb1b4c19202b5 /nn/driver | |
parent | 1c75cf8fe27db074a9f6bb53eef935075473fc2a (diff) | |
download | ml-f65effce2817e390784aa7f2d0c5cf3334694db8.tar.gz |
Update the sample driver to validate unspecified output shape for
executeFenced.
Bug: 148979873
Test: mm
Test: CtsNNAPITestCases
Change-Id: Icbba41cb8c3c027407268244dc6012f610f72e66
Merged-In: Icbba41cb8c3c027407268244dc6012f610f72e66
(cherry picked from commit a1010d4760a32351e97165b29e33365df78af22c)
Diffstat (limited to 'nn/driver')
-rw-r--r-- | nn/driver/sample/SampleDriver.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nn/driver/sample/SampleDriver.cpp b/nn/driver/sample/SampleDriver.cpp index 37009aef7..8bf5a796b 100644 --- a/nn/driver/sample/SampleDriver.cpp +++ b/nn/driver/sample/SampleDriver.cpp @@ -434,7 +434,7 @@ Return<void> SamplePreparedModel::executeFenced( time_point driverStart, driverEnd, deviceStart, deviceEnd; if (measure == MeasureTiming::YES) driverStart = now(); - if (!validateRequest(request, mModel)) { + if (!validateRequest(request, mModel, /*allowUnspecifiedOutput=*/false)) { cb(ErrorStatus::INVALID_ARGUMENT, hidl_handle(nullptr), nullptr); return Void(); } |