diff options
| author | Gregory Chanan <gchanan@fb.com> | 2018-10-30 16:28:30 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-10-30 16:31:53 -0700 |
| commit | f1d02f6d1c3e1c9103e0f75369cba566a3a31fa8 (patch) | |
| tree | 2e41a72ed7be02f52f81200f60ec0d52ce17bf04 /tools/autograd/derivatives.yaml | |
| parent | 11a16961a543a44ad7fc36d144888f5ddf070791 (diff) | |
| download | pytorch-f1d02f6d1c3e1c9103e0f75369cba566a3a31fa8.tar.gz pytorch-f1d02f6d1c3e1c9103e0f75369cba566a3a31fa8.tar.bz2 pytorch-f1d02f6d1c3e1c9103e0f75369cba566a3a31fa8.zip | |
Move underscore prefixed linear algebra TH functions to _th prefix.
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13309
Reviewed By: ezyang
Differential Revision: D12839533
Pulled By: gchanan
fbshipit-source-id: 27bdc5254d2529269b705c2c057826a44297a34b
Diffstat (limited to 'tools/autograd/derivatives.yaml')
| -rw-r--r-- | tools/autograd/derivatives.yaml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tools/autograd/derivatives.yaml b/tools/autograd/derivatives.yaml index 2477a26c6e..2e8af50eb5 100644 --- a/tools/autograd/derivatives.yaml +++ b/tools/autograd/derivatives.yaml @@ -122,12 +122,12 @@ mat1: mm_mat1_backward(grad, mat2, mat1.sizes(), mat1.strides(), alpha) mat2: mm_mat2_backward(grad, mat1, mat2.sizes(), mat2.strides(), alpha) -- name: _addmv(Tensor self, Tensor mat, Tensor vec, *, Scalar beta, Scalar alpha) +- name: _th_addmv(Tensor self, Tensor mat, Tensor vec, *, Scalar beta, Scalar alpha) self: maybe_multiply(grad, beta) mat: grad.ger(vec) * alpha vec: mat.t().mv(grad) * alpha -- name: _addr(Tensor self, Tensor vec1, Tensor vec2, *, Scalar beta, Scalar alpha) +- name: _th_addr(Tensor self, Tensor vec1, Tensor vec2, *, Scalar beta, Scalar alpha) self: maybe_multiply(grad, beta) vec1: grad.mv(vec2) * alpha vec2: grad.t().mv(vec1) * alpha @@ -490,7 +490,7 @@ self: grad.clone().masked_fill_(self >= other, 0) other: grad.clone().masked_fill_(self < other, 0) -- name: _mm(Tensor self, Tensor mat2) +- name: _th_mm(Tensor self, Tensor mat2) self: mm_mat1_backward(grad, mat2, self.sizes(), self.strides(), 1) mat2: mm_mat2_backward(grad, self, mat2.sizes(), mat2.strides(), 1) |