Warning

This page was created from a pull request.

jax.core.ClosedJaxpr¶

class jax.core.ClosedJaxpr(jaxpr, consts)[source]¶

Parameters

jaxpr (Jaxpr) –
consts (Sequence) –

__init__(jaxpr, consts)[source]¶

Initialize self. See help(type(self)) for accurate signature.

Parameters

jaxpr (Jaxpr) –
consts (Sequence) –

Methods

`__init__`(jaxpr, consts)	Initialize self.
`map_jaxpr`(f)

Attributes

`in_avals`
`literals`
`out_avals`