Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/sparse/
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
DSparseCsrTensorMath.cpp907 scalar_t cast_value = alpha.to<scalar_t>(); in add_out_dense_sparse_compressed_cpu() local
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseCsrTensorMath.cu234 scalar_t cast_value = alpha.to<scalar_t>(); in add_out_dense_sparse_compressed_cuda() local