adding type information to missing ops

Author: dboyliao

utensor_cgen/backend/utensor/snippets/rearch/_snippets.py

@@ -237,17 +237,17 @@ class MissingOpEvalSnippet(OpEvalSnippet):
 
   def __init__(self, op_info, tensor_var_map):
     Snippet.__init__(self)
-    input_var_names = [tensor_var_map[tensor.name] for tensor in op_info.input_tensors]
-    out_tensor_names = [tensor.name for tensor in op_info.output_tensors]
-    out_var_names = [tensor_var_map[tensor.name] for tensor in op_info.output_tensors]
+
     quant_params_map = {}
     for out_tensor in op_info.output_tensors:
       quant_params = self.get_quant_param(out_tensor)
       quant_params_map[out_tensor.name] = quant_params
     self.template_vars['op_type'] = op_info.op_type
-    self.template_vars['input_var_names'] = input_var_names
-    self.template_vars['out_var_names'] = out_var_names
-    self.template_vars['out_tensor_names'] = out_tensor_names
+    self.template_vars['input_tensors'] = op_info.input_tensors[:]
+    self.template_vars['out_var_names'] = [
+      tensor_var_map[tensor.name] for tensor in op_info.output_tensors
+    ]
+    self.template_vars['output_tensors'] = op_info.output_tensors[:]
     self.template_vars['quant_params_map'] = quant_params_map
 
 

utensor_cgen/backend/utensor/snippets/templates/snippets/rearch/op_missing.cpp

@@ -2,18 +2,18 @@
     FIXME: {{op_type}} currently not supported, you have to fill up this section or it won't compile
 
     Input Tensors:
-    {%for name in input_var_names%}
-        {{name}}
+    {%for tensor in input_tensors%}
+        {{tensor.name}}, of type {{tensor.dtype}}
     {%endfor%}
 
     Output Tensors:
-    {%for name, var_name in zip(out_tensor_names, out_var_names)%}
-        {{name}} should be named as {{var_name}}
-        {%if quant_params_map[name]%}
+    {%for tensor, var_name in zip(output_tensors, out_var_names)%}
+        {{tensor.name}} is of type {{tensor.dtype}} and should be named as {{var_name}}
+        {%if quant_params_map[tensor.name]%}
             quantization parameters:
-            - zero point: {{quant_params_map[name]["zero_point"]["value"]}}, {{quant_params_map[name]["zero_point"]["type_str"]}}
-            - scale: {{quant_params_map[name]["scale"]["value"]}}, {{quant_params_map[name]["scale"]["type_str"]}}
-            - is per tensor quantization: {{quant_params_map[name]["is_per_tensor"]}}
+            - zero point: {{quant_params_map[tensor.name]["zero_point"]["value"]}}, {{quant_params_map[tensor.name]["zero_point"]["type_str"]}}
+            - scale: {{quant_params_map[tensor.name]["scale"]["value"]}}, {{quant_params_map[tensor.name]["scale"]["type_str"]}}
+            - is per tensor quantization: {{quant_params_map[tensor.name]["is_per_tensor"]}}
         {%endif%}
     {%endfor%}
     */