Commit 7979eff9 authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Only convert BackpropFilterConv to depthwise convolution if format is NHWC. 

The logic for the conversion has this assumption. This CL makes sure that this
assumption holds, and adds tests for NCHW format.

PiperOrigin-RevId: 225534508
parent ab6229b5
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