oryx.experimental.matching.jax_rewrite.Params

An immutable dictionary used to represent parameters of JAX primitives.

Inherits From: Pattern

oryx.experimental.matching.jax_rewrite.Params(
    params: Optional[Dict[str, Any]] = None, **kwargs
)

params an optional dictionary that the Params object will be initialized from.
**kwargs keyword arguments that will populate the Params.

Methods

keys

View source

keys()

match

View source

match(
    expr: Expr,
    bindings: oryx.experimental.matching.jax_rewrite.Bindings,
    succeed: oryx.experimental.matching.jax_rewrite.Continuation
) -> oryx.experimental.matching.jax_rewrite.Success

__contains__

View source

__contains__(
    key
)

__eq__

View source

__eq__(
    other
)

Return self==value.

__getitem__

View source

__getitem__(
    key
)

__iter__

View source

__iter__()