jax.core.ClosedJaxpr

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

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

Parameters

Methods

__init__(jaxpr, consts)

Initialize self.

Attributes

in_avals

literals

out_avals