[CI] Fix MLF input and output name map (#13740)

This PR fixes error introduced by #13704
apache · Jan 10, 2023 · d00168f · d00168f
1 parent 687ec78
commit d00168f
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/python/tvm/micro/model_library_format.py b/python/tvm/micro/model_library_format.py
@@ -468,12 +468,6 @@ def _export_graph_model_library_format(
             input_names = list(inputs.keys())
             output_names = list(outputs.keys())
 
-            input_name_to_size_map = {
-                name: property_map["size"] for name, property_map in inputs_sizes.items()
-            }
-            output_name_to_size_map = {
-                name: property_map["size"] for name, property_map in output_sizes.items()
-            }
             generate_c_interface_header(
                 mod.libmod_name,
                 input_names,
@@ -483,8 +477,8 @@ def _export_graph_model_library_format(
                 devices,
                 workspace_size,
                 include_path,
-                input_name_to_size_map,
-                output_name_to_size_map,
+                inputs_sizes,
+                output_sizes,
             )
 
         is_aot = isinstance(mod, executor_factory.AOTExecutorFactoryModule)