dmx.compressor.fx.transformer.record_input_interpreter.RecordInputInterpreterExport

class dmx.compressor.fx.transformer.record_input_interpreter.RecordInputInterpreterExport(module, garbage_collect_values=True, graph=None)

Interpreter that captures the input tensors to the compound functions. .. attribute:: nodeInputs

dictionary that maps node name to its tensor args and kwargs

__init__(module, garbage_collect_values=True, graph=None)

Note

Backwards-compatibility for this API is guaranteed.

Methods

__init__(module[, garbage_collect_values, graph])

boxed_run(args_list)

Run module via interpretation and return the result.

call_function(target, args, kwargs)

Execute a call_function node and return the result.

call_method(target, args, kwargs)

Execute a call_method node and return the result.

call_module(target, args, kwargs)

Execute a call_module node and return the result.

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 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.