Replies: 1 comment 1 reply
-
Array arguments cannot be static, so you cannot mark To fix this you can either (1) not mark |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hey there! Following this guide: https://openxla.org/stablehlo/tutorials/jax-export I am trying to export stablehlo from a function that takes additional arguments.
Below is an example with convolution.
What should be the correct type of kwargs (
params
in my example)?Given the errors I get (some of them added as comments)
params
should be hash-able but also having shape and type but not only symbolic specs, the real value is needed.Is this the correct way of exporting this kind of functions?
I tried using wrapping
conv
in a lambda and puttingparams
in the body. This approach works but produces different stablehlo output.Beta Was this translation helpful? Give feedback.
All reactions