Commit 5c8a7cb3 authored by James Qin's avatar James Qin Committed by TensorFlower Gardener
Browse files

Revamp cudnn_rnn ops level unittests

* Use absl parameterized.named_tuple decorator
* Test training by comparing outputs and grads of cudnn and tf rnn, instead of comparing cudnn's symbolic and numerical gradients. Much more comfortable tolerance levels.
* Split different rnn type's tests, much easier to read. (though paying a little more cost in coding.)

PiperOrigin-RevId: 221502090
parent 1a4e1372
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment