Skip to content

Commit

Permalink
Update jax/experimental/jax2tf/README.md
Browse files Browse the repository at this point in the history
Co-authored-by: Marc van Zee <marcvanzee@gmail.com>
  • Loading branch information
gnecula and marcvanzee committed Dec 12, 2022
1 parent 49d0346 commit 9068c44
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jax/experimental/jax2tf/README.md
Expand Up @@ -637,7 +637,7 @@ polymorphic_shapes = ["a, 2*a, b"]
polymorphic_shapes = ["a * a, a"]
```

Furthermore, in the native lowering the inputs that are not needed in the computation
Furthermore, when native lowering the inputs that are not needed in the computation
are ignored, so the dimension variables must be derivable only from used inputs.
In the following example, the `x_unused` is not part of the computation so its
input shapes cannot be used for deriving the dimension variables, and you will
Expand Down

0 comments on commit 9068c44

Please sign in to comment.