Commit aba19e41 authored by Sanjoy Das's avatar Sanjoy Das Committed by TensorFlower Gardener
Browse files

Add a (failing) randomized test for FusedBatchNorm training 

I had to add a --tf_xla_reference_device flag so that I could set the reference
device to GPU:0.  The CPU kernel for FusedBatchNorm does not support NCHW.

PiperOrigin-RevId: 229586592
parent b4ae5ba3
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