Commit cf65480
[XlaCallModule] Drop support for dim_args_spec attribute.
This attribute was used to support shape polymorphism in versions
up to and including version 4. Starting on March 28th 2023 with
JAX version 0.4.6 we stopped using this attribute. We are now
beyond the 6 month backward compatibility version and we drop
support for this attribute.
We also increase the minimum supported serialization version to 5.
See https://github.com/google/jax/blob/main/jax/experimental/jax2tf/README.md#native-serialization-versions
PiperOrigin-RevId: 5744502041 parent e7dff2c commit cf65480
1 file changed
+2
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
810 | 810 | | |
811 | 811 | | |
812 | 812 | | |
| 813 | + | |
| 814 | + | |
813 | 815 | | |
814 | 816 | | |
815 | 817 | | |
| |||
0 commit comments