![]() |
Keeps track of variables and their values during propagation.
Inherits From: Pytree
oryx.core.interpreters.propagate.Environment(
cell_type, jaxpr
)
Methods
assert_same_type
assert_same_type(
other_env
) -> None
Raises an error if environments do not have matching Jaxprs.
copy
copy() -> "Environment"
flatten
flatten()
join
join(
other: "Environment"
) -> "Environment"
read
read(
var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell
unflatten
@classmethod
unflatten( data, xs )
write
write(
var: VarOrLiteral,
cell: oryx.core.interpreters.propagate.Cell
) -> oryx.core.interpreters.propagate.Cell
write_subenv
write_subenv(
eqn: oryx.core.interpreters.propagate.Equation
,
subenv: "Environment"
) -> None
__contains__
__contains__(
var: VarOrLiteral
)
__getitem__
__getitem__(
var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell