Commit 5b3b689c authored by Yangzihao Wang's avatar Yangzihao Wang Committed by TensorFlower Gardener
Browse files

Add env-var to specify whether to use CUDNN_BATCHNORM_SPATIAL_PERSISTENT for cudnn batchnorm. 

PiperOrigin-RevId: 185771595
parent 808371b3
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