1 /* 2 * Copyright (C) 2021 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_EXECUTION_H 18 #define ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_EXECUTION_H 19 20 #include <nnapi/IExecution.h> 21 #include <nnapi/Result.h> 22 #include <nnapi/Types.h> 23 #include <nnapi/hal/CommonUtils.h> 24 25 #include "PreparedModel.h" 26 27 #include <memory> 28 #include <utility> 29 #include <vector> 30 31 // See hardware/interfaces/neuralnetworks/utils/README.md for more information on HIDL interface 32 // lifetimes across processes and for protecting asynchronous calls across HIDL. 33 34 namespace android::hardware::neuralnetworks::V1_3::utils { 35 36 class Execution final : public nn::IExecution, public std::enable_shared_from_this<Execution> { 37 struct PrivateConstructorTag {}; 38 39 public: 40 static nn::GeneralResult<std::shared_ptr<const Execution>> create( 41 std::shared_ptr<const PreparedModel> preparedModel, Request request, 42 hal::utils::RequestRelocation relocation, V1_2::MeasureTiming measure, 43 OptionalTimeoutDuration loopTimeoutDuration); 44 45 Execution(PrivateConstructorTag tag, std::shared_ptr<const PreparedModel> preparedModel, 46 Request request, hal::utils::RequestRelocation relocation, 47 V1_2::MeasureTiming measure, OptionalTimeoutDuration loopTimeoutDuration); 48 49 nn::ExecutionResult<std::pair<std::vector<nn::OutputShape>, nn::Timing>> compute( 50 const nn::OptionalTimePoint& deadline) const override; 51 52 nn::GeneralResult<std::pair<nn::SyncFence, nn::ExecuteFencedInfoCallback>> computeFenced( 53 const std::vector<nn::SyncFence>& waitFor, const nn::OptionalTimePoint& deadline, 54 const nn::OptionalDuration& timeoutDurationAfterFence) const override; 55 56 private: 57 const std::shared_ptr<const PreparedModel> kPreparedModel; 58 const Request kRequest; 59 const hal::utils::RequestRelocation kRelocation; 60 const V1_2::MeasureTiming kMeasure; 61 const OptionalTimeoutDuration kLoopTimeoutDuration; 62 }; 63 64 } // namespace android::hardware::neuralnetworks::V1_3::utils 65 66 #endif // ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_EXECUTION_H 67