Commit 102bfdfd authored by David Majnemer's avatar David Majnemer Committed by TensorFlower Gardener
Browse files

[XLA] Separate input and output spatial dimensions for convolution

This lets us reason about input spatial dimensions as distinct from output spatial dimensions. By doing this, it opens up more opportunities for assigning more interesting, different, layouts for the activations and the output.

PiperOrigin-RevId: 177117140
parent 8781d69b
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment