Searched defs:grad_self (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SpmmReduceKernel.cpp | 243 const Tensor& grad_self, in spmm_reduce_backward_input_kernel_impl()
293 const Tensor& grad_self, in spmm_reduce_backward_input_arg_kernel_impl()
466 const Tensor& grad_self, in spmm_reduce_backward_input_kernel()
485 const Tensor& grad_self, in spmm_reduce_backward_input_arg_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorBackward.cpp | 29 Tensor grad_self, grad_other; in matmul_backward_nested() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 396 Tensor grad_self, grad_grad_output, grad_weight; in linear_double_backward() local
4501 Tensor grad_self, grad_input2; in cholesky_solve_backward() local
6821 Tensor grad_self, grad_src; in scatter_reduce_backward() local
6920 Tensor grad_self, grad_src; in index_reduce_backward() local
6997 Tensor grad_self = at::zeros_like(self); in take_backward() local
7160 Tensor grad_self; in values_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Activation.cpp | 735 Tensor grad_self = at::empty({0}, self.options()); in _prelu_kernel_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensorMath.cpp | 1458 Tensor grad_self, grad_other; in _sparse_mm_reduce_impl_backward_sparse_csr_cpu() local
|