Fix dimension numbers in LHS transpose rule for conv_general_dilated.
Fix dimension numbers in LHS transpose rule for conv_general_dilated.

Fixes #380.

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