Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[jax2tf] Remove backwards compatibility shims (jax2tf.shape_poly) #18876

Merged
merged 1 commit into from
Dec 11, 2023
Merged

[jax2tf] Remove backwards compatibility shims (jax2tf.shape_poly) #18876

merged 1 commit into from
Dec 11, 2023

Commits on Dec 11, 2023

  1. [jax2tf] Remove backwards compatibility shims (jax2tf.shape_poly) 

    A while ago we moved the native exporting code out of jax2tf, to
jax.experimental.export, but we left behind shape_poly.py shim for
backwards compatibility. Now we remove this shim.

Replace previous

  from jax.experimental.jax2tf import shape_poly

with

  from jax.experimental.export import shape_poly

PiperOrigin-RevId: 589850402
    @gnecula
    gnecula authored and jax authors committed Dec 11, 2023
    Configuration menu
    Copy the full SHA
    b5f852f View commit details
    Browse the repository at this point in the history