diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-09-12 14:17:17 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-12 14:21:00 -0700 |
| commit | c2b3222ac552e9698968c9a212095dbc8b9ca40b (patch) | |
| tree | 3c516598ec6c343116d06da94832fb68c73846a7 /tensorflow/contrib/optimizer_v2 | |
| parent | 31a55ce8927d2659dd20cc540a9cde8dcae3c036 (diff) | |
Use Eigen::CuboidConvolutionBackwardKernel in Conv3DBackpropFilter.
Instead of multiple primitive Eigen ops in Conv3DBackpropFilter, call directly into Eigen function.
Modest ~10-25% latency improvement and ~10-20% peak memory reduction.
PiperOrigin-RevId: 212701797
Diffstat (limited to 'tensorflow/contrib/optimizer_v2')
0 files changed, 0 insertions, 0 deletions