Increase tolerances in a gradient check in tests. This may not be the correct thing to do, but 2e-6 and 1e-6 seem close enough.
Increase tolerances in a gradient check in tests. This may not be the correct thing to do, but 2e-6 and 1e-6 seem close enough.

PiperOrigin-RevId: 228908963

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor