diff options
Diffstat (limited to 'neuralnetworks/utils/common/test/MockDevice.h')
| -rw-r--r-- | neuralnetworks/utils/common/test/MockDevice.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/neuralnetworks/utils/common/test/MockDevice.h b/neuralnetworks/utils/common/test/MockDevice.h index a9428bccfc..a0fc5c3de1 100644 --- a/neuralnetworks/utils/common/test/MockDevice.h +++ b/neuralnetworks/utils/common/test/MockDevice.h @@ -39,7 +39,9 @@ class MockDevice final : public IDevice { MOCK_METHOD(GeneralResult<SharedPreparedModel>, prepareModel, (const Model& model, ExecutionPreference preference, Priority priority, OptionalTimePoint deadline, const std::vector<SharedHandle>& modelCache, - const std::vector<SharedHandle>& dataCache, const CacheToken& token), + const std::vector<SharedHandle>& dataCache, const CacheToken& token, + const std::vector<TokenValuePair>& hints, + const std::vector<nn::ExtensionNameAndPrefix>& extensionNameToPrefix), (const, override)); MOCK_METHOD(GeneralResult<SharedPreparedModel>, prepareModelFromCache, (OptionalTimePoint deadline, const std::vector<SharedHandle>& modelCache, |