dmx.compressor.fx.transformer.export_transformer.ExportSubstituteTransformer

class dmx.compressor.fx.transformer.export_transformer.ExportSubstituteTransformer(gm: GraphModule, model: Module, scope: str = '', node_inputs: dict = None)

__init__(gm: GraphModule, model: Module, scope: str = '', node_inputs: dict = None)

Note

Backwards-compatibility for this API is guaranteed.

Methods

__init__(gm, model[, scope, node_inputs])
add_submod(name, mod)	this function will try to reuse modules in old_gm if possible
boxed_run(args_list)	Run module via interpretation and return the result.
call_function(target, args, kwargs)
call_method(target, args, kwargs)	Execute a call_method node and return the result.
call_module(target, args, kwargs)
fetch_args_kwargs_from_env(n)	Fetch the concrete values of args and kwargs of node n from the current execution environment.
fetch_attr(target)	Fetch an attribute from the Module hierarchy of self.module.
get_attr(target, args, kwargs)	Execute a get_attr node.
map_nodes_to_values(args, n)	Recursively descend through args and look up the concrete value for each Node in the current execution environment.
merge_graph(subgraph, name, args, kwargs)	This function is to merge subgraph to graph
output(target, args, kwargs)	Execute an output node.
placeholder(target, args, kwargs)	Execute a placeholder node.
run(*args[, initial_env, enable_io_processing])	Run module via interpretation and return the result.
run_node(n)	Run a specific node n and return the result.
transform()	Transform self.module and return the transformed GraphModule.