jax
47df8628 - Fix the problem for tf function return StatefulPartitionedCall during jax2tf.call_tf.

Commit
2 years ago
Fix the problem for tf function return StatefulPartitionedCall during jax2tf.call_tf. PiperOrigin-RevId: 529964653
Author
Committer
Parents
Loading