diff options
| author | Haamed Gheibi <haamed@google.com> | 2022-02-04 13:47:26 -0800 |
|---|---|---|
| committer | Haamed Gheibi <haamed@google.com> | 2022-02-04 13:55:47 -0800 |
| commit | f99b35c293439db0b7436b47b939eb8c7bf21b51 (patch) | |
| tree | 6cd9b0719554809447c845616317cca5409b93ae /neuralnetworks/aidl/utils/test/MockPreparedModel.h | |
| parent | a028272dee9220e6810cbdcfb2328c34f8afe4c2 (diff) | |
| parent | 332dead340bb196c6ba3f6978e8fb53966c74bf7 (diff) | |
Merge TP1A.220120.003
Change-Id: Ie5eba313ee102e452f5f96942ed2f3a7bb4e8f01
Diffstat (limited to 'neuralnetworks/aidl/utils/test/MockPreparedModel.h')
| -rw-r--r-- | neuralnetworks/aidl/utils/test/MockPreparedModel.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/neuralnetworks/aidl/utils/test/MockPreparedModel.h b/neuralnetworks/aidl/utils/test/MockPreparedModel.h index a4ae2b778a..0ed9af9929 100644 --- a/neuralnetworks/aidl/utils/test/MockPreparedModel.h +++ b/neuralnetworks/aidl/utils/test/MockPreparedModel.h @@ -17,6 +17,7 @@ #ifndef ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_AIDL_UTILS_TEST_MOCK_PREPARED_MODEL_H #define ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_AIDL_UTILS_TEST_MOCK_PREPARED_MODEL_H +#include <aidl/android/hardware/neuralnetworks/BnExecution.h> #include <aidl/android/hardware/neuralnetworks/BnPreparedModel.h> #include <android/binder_interface_utils.h> #include <gmock/gmock.h> @@ -41,6 +42,10 @@ class MockPreparedModel final : public BnPreparedModel { (override)); MOCK_METHOD(ndk::ScopedAStatus, configureExecutionBurst, (std::shared_ptr<IBurst> * burst), (override)); + MOCK_METHOD(ndk::ScopedAStatus, createReusableExecution, + (const Request& request, bool measureTiming, int64_t loopTimeoutDuration, + std::shared_ptr<IExecution>* execution), + (override)); }; inline std::shared_ptr<MockPreparedModel> MockPreparedModel::create() { |