![]() |
Module for the unzip function transformation.
Unzip is a function transformation that looks for 'variable' instantiations and pulls out concretized variables for partial evaluation. Primitives that return variables are registered in the unzip_registry.
Unzip returns two functions:
init
- maps inputs to variablesapply
- maps variables and inputs to output
Classes
class UnzipTrace
: Contains logic for handling UnzipTracers when tracing a function.
class UnzipTracer
: Tracer whose state encapsulates if the inputs are keys.
class VariableError
: Raised if unable to unzip a function.
Functions
unzip(...)
: Unzip function transformation.
Other Members | |
---|---|
unzip_registry |
|