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

Use Eigen::CuboidConvolutionBackwardInput in Conv3DBackpropInput. 

Instead of multiple primitive Eigen ops in Conv3DBackpropInput, call directly into the ex-NeuralNetworks module's function CuboidConvolutionBackwardInput.

Modest ~10% latency improvement and ~15-20% peak memory reduction.

PiperOrigin-RevId: 212516586
parent 642a1997
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