Commit ebd26397 authored by Yao Zhang's avatar Yao Zhang Committed by TensorFlower Gardener
Browse files

Do not convert layout for FusedBatchNormGrad if is_training is false (freeze 

mode), since NCHW is not supported on GPU in this case.

PiperOrigin-RevId: 176722850
parent b76620ae
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