diff options
Diffstat (limited to 'tools/autograd/derivatives.yaml')
| -rw-r--r-- | tools/autograd/derivatives.yaml | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tools/autograd/derivatives.yaml b/tools/autograd/derivatives.yaml index e652b2e618..9ec8f9efea 100644 --- a/tools/autograd/derivatives.yaml +++ b/tools/autograd/derivatives.yaml @@ -138,6 +138,15 @@ - name: alias(Tensor self) self: grad +# The two items below are necessary because TensorIterator doesn't work on +# Variables (codegen does not unwrap the input Tensor for all(*) without this +# line). +- name: all(Tensor self) + self: not_implemented("all") + +- name: all(Tensor self, int64_t dim, bool keepdim) + self: not_implemented("all") + - name: as_strided(Tensor self, IntList size, IntList stride, int64_t? storage_offset) self: as_strided_backward(grad, TensorGeometry(self), size, stride, storage_offset) |