Class Jaxpr

constructor

• new Jaxpr(inBinders: Var[], eqns: JaxprEqn[], outs: Atom[]): Jaxpr

  Parameters

  • inBinders: Var[]
  • eqns: JaxprEqn[]
  • outs: Atom[]

  Returns Jaxpr

Readonlyeqns

ReadonlyinBinders

Readonlyouts

flatten

• flatten(): Jaxpr

  Flattens nested Jit in a Jaxpr. Useful for handling jit-of-jit.

  Returns Jaxpr

getHash

• getHash(): bigint

  Gets a hash of this Jaxpr.

  Var identity is not considered in the hash, so two Jaxprs with the same order of assignments and operators but different variable IDs will resolve to the same hash (and toString representation).

  Returns bigint

hash

• hash(state: FpHash): void

  Parameters

  • state: FpHash

  Returns void

pprint

• pprint(): PPrint

  Returns PPrint

simplify

• simplify(): Jaxpr

  Produce a simplified Jaxpr with basic optimizations applied.

  • Trim away unused variables.
  • Fold away *1, *0, or +0 operations against literals.
  • Remove no-op movement operations.

  Returns Jaxpr

toString

• toString(): string

  Returns a string representation of an object.

  Returns string