diff options
Diffstat (limited to 'ArmnnPreparedModel.cpp')
-rw-r--r-- | ArmnnPreparedModel.cpp | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/ArmnnPreparedModel.cpp b/ArmnnPreparedModel.cpp index 7a275af4..e4a8b147 100644 --- a/ArmnnPreparedModel.cpp +++ b/ArmnnPreparedModel.cpp @@ -295,11 +295,14 @@ void ArmnnPreparedModel<HalVersion>::ExecuteWithDummyInputs() } } -// Class template specializations -template class ArmnnPreparedModel<HalVersion_1_0>; +/// +/// Class template specializations +/// -#ifdef ARMNN_ANDROID_NN_V1_1 // Using ::android::hardware::neuralnetworks::V1_1. -template class ArmnnPreparedModel<HalVersion_1_1>; +template class ArmnnPreparedModel<hal_1_0::HalPolicy>; + +#if defined(ARMNN_ANDROID_NN_V1_1) +template class ArmnnPreparedModel<hal_1_1::HalPolicy>; #endif -} // namespace armnn_driver +} // namespace armnn_driver
\ No newline at end of file |