Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensorMath.cpp467 scalar_t cast_value = value.to<scalar_t>(); in add_out_sparse_contiguous() local
601 scalar_t cast_value = value.to<scalar_t>(); in add_dense_sparse_worker_non_hybrid_cpu() local
630 scalar_t cast_value = value.to<scalar_t>(); in add_dense_sparse_worker_hybrid_cpu() local
662 scalar_t cast_value = value.to<scalar_t>(); in add_dense_sparse_worker_non_coalesced_cpu() local
H A DSparseCsrTensorMath.cpp907 scalar_t cast_value = alpha.to<scalar_t>(); in add_out_dense_sparse_compressed_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu234 scalar_t cast_value = alpha.to<scalar_t>(); in add_out_dense_sparse_compressed_cuda() local