diff options
Diffstat (limited to 'neuralnetworks/1.2/utils/src/PreparedModel.cpp')
-rw-r--r-- | neuralnetworks/1.2/utils/src/PreparedModel.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/neuralnetworks/1.2/utils/src/PreparedModel.cpp b/neuralnetworks/1.2/utils/src/PreparedModel.cpp index 71a4ea872b..b209a44eba 100644 --- a/neuralnetworks/1.2/utils/src/PreparedModel.cpp +++ b/neuralnetworks/1.2/utils/src/PreparedModel.cpp @@ -122,10 +122,12 @@ PreparedModel::executeFenced(const nn::Request& /*request*/, nn::GeneralResult<nn::SharedBurst> PreparedModel::configureExecutionBurst() const { auto self = shared_from_this(); - auto fallback = [preparedModel = std::move(self)](const nn::Request& request, - nn::MeasureTiming measure) + auto fallback = [preparedModel = std::move(self)]( + const nn::Request& request, nn::MeasureTiming measure, + const nn::OptionalTimePoint& deadline, + const nn::OptionalDuration& loopTimeoutDuration) -> nn::ExecutionResult<std::pair<std::vector<nn::OutputShape>, nn::Timing>> { - return preparedModel->execute(request, measure, {}, {}); + return preparedModel->execute(request, measure, deadline, loopTimeoutDuration); }; const auto pollingTimeWindow = getBurstControllerPollingTimeWindow(); return ExecutionBurstController::create(kPreparedModel, std::move(fallback), pollingTimeWindow); |