Commit 5d5d9f70 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

In the CUDA path of depthwise_conv2d, extend the fastpath to support tensor... 

In the CUDA path of depthwise_conv2d, extend the fastpath to support tensor widths/heights up to 32x32 (from previously 16x16).

PiperOrigin-RevId: 158765635
parent 8cedce4b
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