Home
last modified time | relevance | path

Searched defs:new_values_acc (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorUtils.h417 Tensor new_values, new_values_acc; variable
435 inline void copy_from_acc_buffer(Tensor& new_values, Tensor& new_values_acc) { in copy_from_acc_buffer()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu494 Tensor new_values_acc = std::get<1>(acc_buffer); in reduce_sparse_csr_dim0_cuda_template() local
585 Tensor new_values_acc = std::get<1>(acc_buffer); in reduce_sparse_csr_dim1_cuda_template() local
635 Tensor new_values, new_values_acc; in reduce_sparse_csr_dim01_cuda_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp1080 Tensor new_values_acc = std::get<1>(acc_buffer); in reduce_sparse_csr_dim0_cpu_template() local
1168 Tensor new_values_acc = std::get<1>(acc_buffer); in reduce_sparse_csr_dim1_cpu_template() local