Commit 815f0b32 authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Pad the channel dims of f16 convs to multiples of 8 on Volta. 

This lets us actually use Volta's tensor cores for such convs.

PiperOrigin-RevId: 206649341
parent c57870c1
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