diff options
Diffstat (limited to 'ArmnnPreparedModel_1_3.cpp')
-rw-r--r-- | ArmnnPreparedModel_1_3.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ArmnnPreparedModel_1_3.cpp b/ArmnnPreparedModel_1_3.cpp index 5a370321..63918725 100644 --- a/ArmnnPreparedModel_1_3.cpp +++ b/ArmnnPreparedModel_1_3.cpp @@ -492,9 +492,8 @@ Return<V1_3::ErrorStatus> ArmnnPreparedModel_1_3<HalVersion>::PrepareMemoryForOu size_t bufferSize = 0; #if !defined(ARMNN_ANDROID_S) bufferSize = memPools.at(outputArg.location.poolIndex).getHidlMemory().size(); - if (bufferSize < outputSize) #else - bufferSize = memPools.at(outputArg.location.poolIndex).getMemory().size; + bufferSize = memPools.at(outputArg.location.poolIndex).getSize(); #endif if (bufferSize < outputSize) { |