Allow ConvDimensionNumbers to be passed into conv_transpose (#2915)
Allow ConvDimensionNumbers to be passed into conv_transpose (#2915)


tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax,google/jax