Python ExprMutator.__init__ Examples

Example #1

 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

Example #2

File: _partition_conversions.py Project: aiblackmaner/tvm

 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 = {}

Example #3

File: tensorrt.py Project: vinx13/tvm

 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

Example #4

File: _partition_conversions.py Project: aiblackmaner/tvm

 def __init__(self, quantized_dtypes):
     ExprMutator.__init__(self)
     self.mid_body = None
     self.quantized_dtypes = quantized_dtypes

Example #5

File: test_pass_remove_standalone_reshapes.py Project: chenghanpeng/tvm

 def __init__(self):
     ExprMutator.__init__(self)