Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu575 const bool sum_all_sparse_dim = (input_sparse_dim == sparse_dims_to_sum_size); in _sparse_sum_backward_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp1666 const bool sum_all_sparse_dim = (sparse_dim == sparse_dims_to_sum_size); in _sparse_sum() local
1794 const bool sum_all_sparse_dim = (input_sparse_dim == sparse_dims_to_sum_size); in _sparse_sum_backward_cpu() local