jax
d607164d - make_jaxpr return_shape use ShapeDtypeStruct, test

Commit

5 years ago

make_jaxpr return_shape use ShapeDtypeStruct, test

References

#4364 - Adding an option to return the output tree in make_jaxpr

Author

mattjj

mattjj

Parents

Loading