Ejemplo n.º 1
0
 def __init__(
     self, subgraphs_to_remove: List[str], mod: tvm.IRModule, new_mod: tvm.IRModule
 ) -> None:
     ExprMutator.__init__(self)
     self.subgraphs_to_remove = subgraphs_to_remove
     self.mod = mod
     self.new_mod = new_mod
Ejemplo n.º 2
0
 def __init__(self, params, quantized_dtypes):
     ExprMutator.__init__(self)
     self.params = set(params)
     self.quantized_dtypes = quantized_dtypes
     self.subtree_params = set()
     self.new_func_params = []
     self.prefix_sb = relay.ScopeBuilder()
     self.prefix_binding_map = {}
Ejemplo n.º 3
0
 def __init__(self, subgraphs_to_remove, mod, new_mod):
     ExprMutator.__init__(self)
     self.subgraphs_to_remove = subgraphs_to_remove
     self.mod = mod
     self.new_mod = new_mod
Ejemplo n.º 4
0
 def __init__(self, quantized_dtypes):
     ExprMutator.__init__(self)
     self.mid_body = None
     self.quantized_dtypes = quantized_dtypes
 def __init__(self):
     ExprMutator.__init__(self)