diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-07-05 12:06:04 +0100 |
---|---|---|
committer | surmeh01 <surabhi.mehta@arm.com> | 2018-07-05 12:06:04 +0100 |
commit | deb3bdbe028a59da0759dd7a560387d03a11d322 (patch) | |
tree | 869b7ee10d8f1f19a0861e0b552bb453330adf0a /ArmnnPreparedModel.cpp | |
parent | 49b9e100bfbb3b8da01472a0ff48b2bd92944e01 (diff) | |
download | android-nn-driver-deb3bdbe028a59da0759dd7a560387d03a11d322.tar.gz |
Release 18.05.02
Diffstat (limited to 'ArmnnPreparedModel.cpp')
-rw-r--r-- | ArmnnPreparedModel.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/ArmnnPreparedModel.cpp b/ArmnnPreparedModel.cpp index 1bd72199..3aad955b 100644 --- a/ArmnnPreparedModel.cpp +++ b/ArmnnPreparedModel.cpp @@ -12,6 +12,12 @@ #include <log/log.h> #include <OperationsUtils.h> +#if defined(ARMNN_ANDROID_P) +// The headers of the ML framework have changed between Android O and Android P. +// The validation functions have been moved into their own header, ValidateHal.h. +#include <ValidateHal.h> +#endif + #include <cassert> #include <cinttypes> @@ -101,7 +107,7 @@ void ArmnnPreparedModel::DumpTensorsIfRequired(char const* tensorNamePrefix, ArmnnPreparedModel::ArmnnPreparedModel(armnn::NetworkId networkId, armnn::IRuntime* runtime, - const Model& model, + const V1_0::Model& model, const std::string& requestInputsAndOutputsDumpDir) : m_NetworkId(networkId) , m_Runtime(runtime) @@ -269,7 +275,7 @@ void ArmnnPreparedModel::ExecuteWithDummyInputs() } } -AndroidNnCpuExecutorPreparedModel::AndroidNnCpuExecutorPreparedModel(const Model& model, +AndroidNnCpuExecutorPreparedModel::AndroidNnCpuExecutorPreparedModel(const V1_0::Model& model, const std::string& requestInputsAndOutputsDumpDir) : m_Model(model) , m_RequestInputsAndOutputsDumpDir(requestInputsAndOutputsDumpDir) |