jax
2e35f25b - Consolidate the code path for both `call_tf_graph=True` or `call_tf_graph=False`.

Commit

2 years ago

Consolidate the code path for both `call_tf_graph=True` or `call_tf_graph=False`. PiperOrigin-RevId: 552605585

Author

maxwillzq

Committer

a-googler

Parents