diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
index e72b3e5873c..135d99ebd7f 100644
--- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
+++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
@@ -1726,9 +1726,13 @@ NNAPIDelegateKernel::MappingFn NNAPIDelegateKernel::Map(
           const bool hybrid_op =
               IsHybridOperator(mapping_args.context,
                                kTfLiteBuiltinTransposeConv, mapping_args.node);
-          mapping_args.builder->AddTensorInput(/*kDataInputTensor*/ 2,
-                                               hybrid_op);
-          mapping_args.builder->AddTensorInput(/*kWeightsTensor*/ 1, hybrid_op);
+          mapping_args.builder->AddTensorInput(
+              mapping_args.node->inputs->data[/* kDataInputTensor */ 2],
+              hybrid_op);
+
+          mapping_args.builder->AddTensorInput(
+              mapping_args.node->inputs->data[/* kWeightsTensor */ 1],
+              hybrid_op);
 
           // NNAPI requires a bias tensor, so we allocate a new tensor to fill
           // it with zeroes. It is deleted with other tensors in the context
@@ -1787,8 +1791,9 @@ NNAPIDelegateKernel::MappingFn NNAPIDelegateKernel::Map(
                 /*zero_point=*/0);
           }
 
-          mapping_args.builder->AddTensorInput(/*kOutputShapeTensor*/ 0,
-                                               hybrid_op);
+          mapping_args.builder->AddTensorInput(
+              mapping_args.node->inputs->data[/* kOutputShapeTensor */ 0],
+              hybrid_op);
 
           auto builtin = reinterpret_cast<TfLiteTransposeConvParams*>(
               mapping_args.node->builtin_data);