Home
last modified time | relevance | path

Searched refs:cpu_sparse_coo_softmax_backward (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSoftMax.cpp392 void cpu_sparse_coo_softmax_backward(const Tensor& grad_input, const Tensor& grad, const Tensor& ou… in cpu_sparse_coo_softmax_backward() function
583 cpu_sparse_coo_softmax_backward<scalar_t, false>( in softmax_backward_sparse_cpu()
603 cpu_sparse_coo_softmax_backward<scalar_t, true>( in log_softmax_backward_sparse_cpu()