Home
last modified time | relevance | path

Searched defs:grad_input_values (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu581 auto grad_input_values = grad_; in _sparse_sum_backward_cuda() local
608 Tensor grad_input_values; in _sparse_sum_backward_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp1800 auto grad_input_values = grad_; in _sparse_sum_backward_cpu() local
1829 Tensor grad_input_values; in _sparse_sum_backward_cpu() local