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

Shuffle output_sizes when testing NCHW. 

The output shape needs to reflect that the input has been transposed to NCHW.
Without this fix the test fails with XLA somewhere inside
compute_gradient_error.

I have not done the analysis to figure out how this test passes with TF without
XLA.

PiperOrigin-RevId: 223895440
parent 697339bc
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