Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/sparse/
DSoftMax.cpp492 auto grad_values_row = grad_values_accessor[j]; in cpu_sparse_coo_softmax_backward() local
511 auto grad_values_row = grad_values_accessor[j]; in cpu_sparse_coo_softmax_backward() local
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSoftMax.cu191 auto grad_values_row = grad_values_accessor[j]; in cuda_sparse_coo_softmax_backward_kernel() local
207 auto grad_values_row = grad_values_accessor[j]; in cuda_sparse_coo_softmax_backward_kernel() local