Converts a JAX value type into a shaped abstract value.
View aliases
Main aliases
oryx.experimental.matching.jax_rewrite.trace_util.get_shaped_aval
oryx.core.trace_util.get_shaped_aval(
x
)
Converts a JAX value type into a shaped abstract value.
Main aliases
oryx.experimental.matching.jax_rewrite.trace_util.get_shaped_aval
oryx.core.trace_util.get_shaped_aval(
x
)
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-05-23 UTC.