Fix wrap_jax when there is a mix of statically known and unknown shapes
#5476
This job was skipped
Loading
wrap_jax when there is a mix of statically known and unknown shapes
#5476