oryx.core.interpreters.propagate.Environment

Keeps track of variables and their values during propagation.

Inherits From: Pytree

Methods

assert_same_type

View source

Raises an error if environments do not have matching Jaxprs.

copy

View source

flatten

View source

join

View source

read

View source

unflatten

View source

write

View source

write_subenv

View source

__contains__

View source

__getitem__

View source