pytorch
f135438d - Dispatch to at::convolution intead of at::_convolution in _convolution_double_backward (#70661)

Commit
4 years ago
Dispatch to at::convolution intead of at::_convolution in _convolution_double_backward (#70661) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70661 Dispatching to at::convolution can make Lazy Tensor trace the right convolution op. Test Plan: pytest test/test_nn.py -k test_conv_double_backward_strided_with_3D_input_and_weight Reviewed By: wconstab, jbschlosser Differential Revision: D33428780 Pulled By: desertfire fbshipit-source-id: 899e4135588ea33fff23d16103c25d9bcd3f902c
Author
Parents
Loading