Skip to content

Commit

Permalink
[jax2tf] Refactor the backwards compatibility tests.
Browse files Browse the repository at this point in the history 
We have two goals:

   * separate out the tests that depend on TensorFlow (back_compat_tf_tests.py)
   * separate out the boilerplate (back_compat_test_util.py) so that we can use it as a library to set up similar compatibility tests for custom calls that are not used directly by the JAX core.

PiperOrigin-RevId: 542500216
  • Loading branch information
@gnecula
gnecula authored and jax authors committed Jun 22, 2023
1 parent f02d122 commit c5a47d1
Show file tree
Hide file tree
Showing 3 changed files with 451 additions and 381 deletions.
Loading

0 comments on commit c5a47d1

Please sign in to comment.