Adjust tolerance for LaxTest.testConv0DIsDot. (#1978)
Adjust tolerance for LaxTest.testConv0DIsDot. (#1978)

This was failing on TPU.
google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax