Commit f361fb8e authored by Tim Shen's avatar Tim Shen Committed by TensorFlower Gardener
Browse files

Further simplify the cuDNN wrappers. Instead of passing around 

CudnnConvParams, just pass around the HloInstruction.

This is based on the observation that most code doesn't care about the
convolution semantics like which operand is input vs filter vs output.
In fact, only layout assignment and conv runner care about them.

PiperOrigin-RevId: 214307399
parent 28eeda83
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