android-nn-driver: size access API accommodation The size access had a minor API delta, accomodate this. Test: build Change-Id: Icf77d80f8b7c376aefc7fb53b60f9cd2d340aa78
diff --git a/ArmnnPreparedModel_1_2.cpp b/ArmnnPreparedModel_1_2.cpp index 689634e..b37ac24 100644 --- a/ArmnnPreparedModel_1_2.cpp +++ b/ArmnnPreparedModel_1_2.cpp
@@ -265,7 +265,7 @@ return V1_0::ErrorStatus::OUTPUT_INSUFFICIENT_SIZE; } - const size_t bufferSize = memPools.at(outputArg.location.poolIndex).getHidlMemory().size(); + const size_t bufferSize = memPools.at(outputArg.location.poolIndex).getSize(); if (bufferSize < outputSize) { ALOGW("ArmnnPreparedModel_1_2::Execute failed: bufferSize < outputSize");
diff --git a/ArmnnPreparedModel_1_3.cpp b/ArmnnPreparedModel_1_3.cpp index 7538490..d33873f 100644 --- a/ArmnnPreparedModel_1_3.cpp +++ b/ArmnnPreparedModel_1_3.cpp
@@ -478,7 +478,7 @@ return V1_3::ErrorStatus::OUTPUT_INSUFFICIENT_SIZE; } - const size_t bufferSize = memPools.at(outputArg.location.poolIndex).getHidlMemory().size(); + const size_t bufferSize = memPools.at(outputArg.location.poolIndex).getSize(); if (bufferSize < outputSize) { ALOGW("ArmnnPreparedModel_1_3::Execute failed bufferSize (%s) < outputSize (%s)",