jax
a12e5d56 - Fix export and typed prng key. While lowering call_exported, if the out_avals were a typed prng key, it was not being converted to physical avals before lowering.

Commit

136 days ago

Fix export and typed prng key. While lowering call_exported, if the out_avals were a typed prng key, it was not being converted to physical avals before lowering. PiperOrigin-RevId: 923683701

References

#38076 - Fix export and typed prng key. While lowering call_exported, if the out_avals were a typed prng key, it was not being converted to physical avals before lowering.

Author

yashk2810

Committer

Google-ML-Automation

Parents

66e0fd37

jax a12e5d56 - Fix export and typed prng key. While lowering call_exported, if the out_avals were a typed prng key, it was not being converted to physical avals before lowering.

jax
a12e5d56 - Fix export and typed prng key. While lowering call_exported, if the out_avals were a typed prng key, it was not being converted to physical avals before lowering.