Commit 0341c5ae authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Don't assume MaybePaddedAndSlicedInput moved the padding into a kPad.

CudnnConvPaddingLegalization has logic to make sure that we call cuDNN with
the same low and high padding, and no interior padding (dilation). However it
then resets the padding in the window, even in the case when this padding was
not moved to a kPad. This CL now moves the update of the convolution window to
the same function that creates the kPad.

PiperOrigin-RevId: 224325347
parent 60a23f7e
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment