Searched refs:cpu_sparse_coo_softmax (Results 1 – 1 of 1) sorted by relevance
158 void cpu_sparse_coo_softmax(Tensor output, const Tensor& input, const int64_t dim) { in cpu_sparse_coo_softmax() function547 cpu_sparse_coo_softmax<scalar_t, false>(output, input, dim); in softmax_sparse_cpu()564 cpu_sparse_coo_softmax<scalar_t, true>(output, input, dim); in log_softmax_sparse_cpu()