Commit ffd9519c authored by Eugene Zhulenev's avatar Eugene Zhulenev Committed by TensorFlower Gardener
Browse files

Optimize CuboidConvolutionBackwardKernel (Conv3D kernel backprop). 

 * simplify contraction by collapsing inner dims into single dimension
 * get rid of expensive reverse op

~5X improvement when compiled with AVX.

PiperOrigin-RevId: 211518363
parent 0cd9b3e4
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