Fix typo in ShapeDtypeStruct (#2253)
Fix typo in ShapeDtypeStruct (#2253)

* fix ShapeDtypeSTruct dtype bug

* move dtype conversion to constructor
tensorflow/probability,google/jax,google/jax,google/jax,tensorflow/probability,google/jax