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

Contraction mapper for cuboid convolutions. 

Directly pack rhs memory for the gebp kernes with a gemm_pack_rhs specialization. It's similar to optimized memory packing in eigen_spatial_convolutions.

Works for:
 1. CuboidConvolution
 2. CuboidConvolutionBackwardInput

~2x-4x speedup when compiled with AVX (depends on tensor&patch dimensions).

PiperOrigin-RevId: 212488060
parent 6cb9189c
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