Making the bias in the convolution module use the dtype of the inputs, the same way as the weights.
Making the bias in the convolution module use the dtype of the inputs, the same way as the weights.

PiperOrigin-RevId: 298299105
Change-Id: I61fcd9b2b1220dc8ad8f77554052d483586592a3

deepmind/dm-haiku,deepmind/dm-haiku