Fix bug and remove unneed header

Author: dboyliao

utensor_cgen/backend/utensor/code_generator/rearch/_code_generator.py

@@ -150,14 +150,15 @@ def _try_get_opt_snippets(self, ugraph, ops, tensor_var_map):
           )
           visited_tensors.add(tensor_info)
         else:
-          for tensor_info in op_info.output_tensors:
-            local_snippets.append(
-              DeclareRamTensorSnippet(
-                tensor_info=tensor_info,
-                tensor_var=tensor_var_map[tensor_info.name]
+          if op_info.name not in ugraph.output_nodes:
+            for tensor_info in op_info.output_tensors:
+              local_snippets.append(
+                DeclareRamTensorSnippet(
+                  tensor_info=tensor_info,
+                  tensor_var=tensor_var_map[tensor_info.name]
+                )
               )
-            )
-            visited_tensors.add(tensor_info)
+              visited_tensors.add(tensor_info)
           op = OperatorFactory.get_opertor(op_info)
           op_name = ops_map[op]
           local_snippets.append(

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

@@ -225,7 +225,7 @@ class QuantizedFullyConnectedSnippet(OpEvalSnippet):
 
 class TimeSlotContainer(SnippetBase):
   __template_name__ = 'containers/rearch/time_slot.cpp'
-  __headers__ = set(['"uTensor.h"', "<vector>"])
+  __headers__ = set(['"uTensor.h"'])
 
   def __init__(self ):
     SnippetBase.__init__(self)
@@ -247,7 +247,7 @@ def render(self):
 
 class SimpleContainer(SnippetBase):
   __template_name__ = 'containers/rearch/simple.cpp'
-  __headers__ = set(['"uTensor.h"', "<vector>"])
+  __headers__ = set(['"uTensor.h"'])
 
   def __init__(self):
     SnippetBase.__init__(self)