namespace fix

Author: dboyliao

utensor_cgen/backend/utensor/_graph_lower/_op_lower.py

@@ -54,6 +54,10 @@ def apply(cls, ugraph):
           op_info.op_type = 'QuantizedDepthwiseSeparableConvOperator'
         for op_info in ugraph.get_ops_by_type('FullyConnectedOperator'):
           op_info.op_type = 'QuantizedFullyConnectedOperator'
+      for op_info in ugraph.get_ops_by_type('DequantizeOperator'):
+        op_info.code_gen_attributes['namespaces'] = ('TFLM',)
+      for op_info in ugraph.get_ops_by_type('QuantizeOperator'):
+        op_info.code_gen_attributes['namespaces'] = ('TFLM',)
 
     @classmethod
     def _check_quantized(cls, ugraph):

utensor_cgen/backend/utensor/code_generator/rearch/_operators/_impls.py

@@ -123,13 +123,14 @@ def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
 @OperatorFactory.register
 class _QuantizeOperator(_Operator):
   op_type = "QuantizeOperator"
+  namespaces = ('TFLM',)
 
   def get_declare_snippet(self, op_var_name, tensor_var_map):
     return DeclareOpSnippet(
       op=self,
       templ_dtypes=[self.out_dtypes[0], self.in_dtypes[0]],
       op_var_name=op_var_name,
-      nested_namespaces=['TFLM'],
+      nested_namespaces=self.namespaces,
     )
 
   def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
@@ -138,20 +139,21 @@ def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
       templ_dtypes=[self.out_dtypes[0], self.in_dtypes[0]],
       op_name=op_var_name,
       tensor_var_map=tensor_var_map,
-      nested_namespaces=['TFLM'],
+      nested_namespaces=self.namespaces,
     )
 
 
 @OperatorFactory.register
 class _DequantizeOperator(_Operator):
   op_type = "DequantizeOperator"
+  namespaces = ('TFLM',)
 
   def get_declare_snippet(self, op_var_name, tensor_var_map):
     return DeclareOpSnippet(
       op=self,
       templ_dtypes=[self.out_dtypes[0], self.in_dtypes[0]],
       op_var_name=op_var_name,
-      nested_namespaces=['TFLM'],
+      nested_namespaces=self.namespaces,
     )
 
   def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
@@ -160,7 +162,7 @@ def get_eval_snippet(self, op_var_name, op_info, tensor_var_map):
       templ_dtypes=[self.out_dtypes[0], self.in_dtypes[0]],
       op_name=op_var_name,
       tensor_var_map=tensor_var_map,
-      nested_namespaces=['TFLM'],
+      nested_namespaces=self.namespaces,
     )
 
 

utensor_cgen/cli/backend.py

@@ -43,8 +43,8 @@ def list_trans_methods(verbose):
 
 @cli.command(name='list-support-ops', help='list all supported op in the backend')
 @click.help_option('-h', '--help')
-@click.option('--target', default='utensor')
-@click.option('--config', default='utensor_cli.toml')
+@click.option('--target', default='utensor', show_default=True)
+@click.option('--config', default='utensor_cli.toml', show_default=True)
 def list_support_ops(target, config):
   from utensor_cgen.backend.api import BackendManager
   if os.path.exists(config):