diff options
Diffstat (limited to 'neuralnetworks/1.0/utils/src/Burst.cpp')
| -rw-r--r-- | neuralnetworks/1.0/utils/src/Burst.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/neuralnetworks/1.0/utils/src/Burst.cpp b/neuralnetworks/1.0/utils/src/Burst.cpp index 971ad08015..e3a97579cd 100644 --- a/neuralnetworks/1.0/utils/src/Burst.cpp +++ b/neuralnetworks/1.0/utils/src/Burst.cpp @@ -20,6 +20,7 @@ #include <nnapi/IBurst.h> #include <nnapi/IPreparedModel.h> #include <nnapi/Result.h> +#include <nnapi/TypeUtils.h> #include <nnapi/Types.h> #include <memory> @@ -48,8 +49,10 @@ Burst::OptionalCacheHold Burst::cacheMemory(const nn::SharedMemory& /*memory*/) } nn::ExecutionResult<std::pair<std::vector<nn::OutputShape>, nn::Timing>> Burst::execute( - const nn::Request& request, nn::MeasureTiming measure) const { - return kPreparedModel->execute(request, measure, {}, {}); + const nn::Request& request, nn::MeasureTiming measure, + const nn::OptionalTimePoint& deadline, + const nn::OptionalDuration& loopTimeoutDuration) const { + return kPreparedModel->execute(request, measure, deadline, loopTimeoutDuration); } } // namespace android::hardware::neuralnetworks::V1_0::utils |