oryx.core.interpreters.propagate.Environment

View source on GitHub

Keeps track of variables and their values during propagation.

oryx.core.interpreters.propagate.Environment(
    cell_type, jaxpr
)

Methods

read

View source

read(
    var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell

read_state

View source

read_state(
    eqn: oryx.core.interpreters.propagate.Equation
) -> State

write

View source

write(
    var: VarOrLiteral,
    cell: oryx.core.interpreters.propagate.Cell
) -> oryx.core.interpreters.propagate.Cell

write_state

View source

write_state(
    eqn: oryx.core.interpreters.propagate.Equation,
    state: State
) -> None

__contains__

View source

__contains__(
    var: VarOrLiteral
)

__getitem__

View source

__getitem__(
    var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell