minor fix

apache · Mar 6, 2024 · 4b3a349 · 4b3a349
1 parent 723452c
commit 4b3a349
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/contrib/msc/framework/tensorflow/runtime/runner.py b/python/tvm/contrib/msc/framework/tensorflow/runtime/runner.py
@@ -156,7 +156,7 @@ def _call_runnable(
         """
 
         input_names = [i["name"] for i in self.get_inputs()]
-        feed_dict = {i + ":0": msc_utils.cast_array(inputs[i["name"]]) for i in input_names}
+        feed_dict = {i + ":0": msc_utils.cast_array(inputs[i]) for i in input_names}
         return runnable.run(self._tf_outputs, feed_dict)
 
     def _device_enabled(self, device: str) -> bool: