jax
47df8628 - Fix the problem for tf function return StatefulPartitionedCall during jax2tf.call_tf.
Commit
2 years ago
References
Author
Committer
Parents