Commit 7d5ef917 authored by Tim Shen's avatar Tim Shen Committed by TensorFlower Gardener
Browse files

For cuDNN >= 7.4, undo the work-around for NHWC with >2 strides for conv backward data. 

In cuDNN 7.4 release notes, they fixed the performance issue that we
work-arounded, and it improves resnet50 a lot.

PiperOrigin-RevId: 223100593
parent 3831f175
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment