def _interpret_jaxpr(jaxpr: core.ClosedJaxpr, *args: TfVal) -> Sequence[TfVal]:
    'Evaluates a Jaxpr with tf.Tensor arguments.\n\n  The output is a sequence of TfVal (no `core.unit`), suitable for use with TF.\n  '
    fun: lu.WrappedFun = lu.wrap_init(core.jaxpr_as_fun(jaxpr))
    out_with_avals = _interpret_fun(fun, args, jaxpr.in_avals)
    return tuple((v for (v, _) in out_with_avals))
Evaluates a Jaxpr with tf.Tensor arguments.

The output is a sequence of TfVal (no `core.unit`), suitable for use with TF.
def _interpret_jaxpr(jaxpr: core.ClosedJaxpr, *args: TfVal) -> Sequence[TfVal]:
    'Evaluates a Jaxpr with tf.Tensor arguments.\n\n  The output is a sequence of TfVal (no `core.unit`), suitable for use with TF.\n  '
    fun: lu.WrappedFun = lu.wrap_init(core.jaxpr_as_fun(jaxpr))
    out_with_avals = _interpret_fun(fun, args, jaxpr.in_avals)
    return tuple((v for (v, _) in out_with_avals))