diff options
-rw-r--r-- | shim_and_sl/ShimDevice.cpp | 2 |
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; |