dmx.compressor.fx.transformer.nodedict_transformer.NodeDictTransformer

class dmx.compressor.fx.transformer.nodedict_transformer.NodeDictTransformer(module: GraphModule)

A transformer that creates a dict contaning mapping between target of node and the node itself

Parameters:

module (fx.GraphModule) – the module to create the mapping

module

the module to create the mapping

Type:

fx.GraphModule

nodeDict

dictionary to store the mapping between occurance order of the node and the node

Type:

dict

__init__(module: GraphModule)

Note

Backwards-compatibility for this API is guaranteed.

Methods

__init__(module)
boxed_run(args_list)	Run module via interpretation and return the result.
call_function(target, args, kwargs)	Execute a call_function node.
call_method(target, args, kwargs)	Execute a call_method node.
call_module(target, args, kwargs)	Execute a call_module node.
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.
output(target, args, kwargs)	Execute a ouptput 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()	Run module via interpretation and return the the nodeDict.