Fix ShimDevice::allocate, wrong operand used for output role.android-vts-12.1_r2android-cts-12.1_r2android-12.1.0_r9android-12.1.0_r8android-12.1.0_r7android-12.1.0_r22android-12.1.0_r21android-12.1.0_r20android-12.1.0_r19android-12.1.0_r11android-12.1.0_r10android12L-devandroid12-qpr3-s7-releaseandroid12-qpr3-s6-releaseandroid12-qpr3-s5-releaseandroid12-qpr3-s4-releaseandroid12-qpr3-s3-releaseandroid12-qpr3-s2-releaseandroid12-qpr3-s1-releaseandroid12-qpr3-release

Bug: 210970637
Test: cts
Change-Id: I76d5b89e2d95d4e775ca544c54ec5100e404ba47
Merged-In: I76d5b89e2d95d4e775ca544c54ec5100e404ba47
(cherry picked from commit b5c6b2dca691228341b20e2ac39c1c7e975d954e)

1 files changed, 1 insertions, 1 deletions

diff --git a/shim_and_sl/ShimDevice.cpp b/shim_and_sl/ShimDevice.cpp
index 06e18d7d2..eadbbef4f 100644
--- a/shim_and_sl/ShimDevice.cpp
+++ b/shim_and_sl/ShimDevice.cpp
@@ -416,7 +416,7 @@ class ShimBuffer : public BnBuffer {
                              "ShimDriver::allocate -- ANeuralNetworksMemoryDesc_addInputRole fail");
         }
         const auto& model = pmodel->getMainModel();
-        const auto& op = model.getOperands()[model.getInputs()[role.ioIndex]];
+        const auto& op = model.getOperands()[model.getOutputs()[role.ioIndex]];
         auto operandType = static_cast<::android::nn::OperandType>(op.operandType.type);
         if (!type) {
             type = operandType;