From d1b6835fec5ec3ec8e2a0794dbfc0894723575d4 Mon Sep 17 00:00:00 2001
From: Hansong Zhang <hsz@meta.com>
Date: Wed, 24 Apr 2024 21:58:46 -0700
Subject: [PATCH] Eliminate deprecated api usage (#2695)

Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/2695

Reviewed By: mergennachin

Differential Revision: D55091814
---
 runtime/executor/method.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/runtime/executor/method.cpp b/runtime/executor/method.cpp
index 59a00ebd3af..378adf15288 100644
--- a/runtime/executor/method.cpp
+++ b/runtime/executor/method.cpp
@@ -786,7 +786,7 @@ Method::set_input(const EValue& input_evalue, size_t input_idx) {
       input_idx,
       inputs_size());
 
-  const auto& e = get_input(input_idx);
+  const auto& e = get_value(get_input_index(input_idx));
   ET_CHECK_OR_RETURN_ERROR(
       e.isTensor() || e.isScalar(),
       InvalidArgument,
@@ -946,7 +946,7 @@ Method::set_output_data_ptr(void* buffer, size_t size, size_t output_idx) {
       output_idx,
       outputs_size());
 
-  auto& output = mutable_output(output_idx);
+  auto& output = mutable_value(get_output_index(output_idx));
   ET_CHECK_OR_RETURN_ERROR(
       output.isTensor(),
       InvalidArgument,