Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseBlasLegacy.cpp21 scalar_t cast_alpha = alpha.to<scalar_t>(); in s_addmm_out_csr_sparse_dense_cuda_worker() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp1189 scalar_t cast_alpha = alpha.to<scalar_t>(); in s_addmm_out_sparse_dense_worker() local
1566 scalar_t cast_alpha = alpha.to<scalar_t>(); in _sspaddmm_out_cpu() local
H A DSparseCsrTensorMath.cpp538 scalar_t cast_alpha = alpha.to<scalar_t>(); in addmm_out_sparse_csr_native_cpu() local