Searched defs:commonDtype (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 261 auto commonDtype = at::result_type(self, value); in div_sparse() local
274 auto commonDtype = at::result_type(self, value); in div_sparse() local
336 auto commonDtype = at::result_type(self, value); in floor_divide_sparse() local
411 auto commonDtype = at::result_type(self, other); in add_sparse() local
439 …& r, const SparseTensor& t, const SparseTensor& src, const Scalar& value, ScalarType commonDtype) { in add_out_sparse_contiguous()
527 …& r, const SparseTensor& t, const SparseTensor& src, const Scalar& value, ScalarType commonDtype) { in add_out_sparse_non_contiguous()
568 auto commonDtype = promoteTypes(t.scalar_type(), src.scalar_type()); in add_out_sparse_cpu() local
722 auto commonDtype = promoteTypes(dense.scalar_type(), sparse_.scalar_type()); in add_out_dense_sparse_cpu() local
804 auto commonDtype = at::result_type(self, other); in mul_sparse() local
1112 auto commonDtype = promoteTypes(t_.scalar_type(), src_.scalar_type()); in mul_out_sparse_cpu() local
|
| H A D | SparseCsrTensorMath.cpp | 269 auto commonDtype = at::result_type(self, other); in mul_sparse_csr() local
783 auto commonDtype = at::result_type(self, other); in add_sparse_csr() local
842 auto commonDtype = promoteTypes(dense.scalar_type(), src.scalar_type()); in add_out_dense_sparse_compressed_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDATensorMath.cu | 288 auto commonDtype = at::result_type(dense, sparse); in add_out_dense_sparse_cuda() local
390 auto commonDtype = at::result_type(t, src); in add_out_sparse_cuda() local
|
| H A D | SparseCsrTensorMath.cu | 171 auto commonDtype = promoteTypes(dense.scalar_type(), src.scalar_type()); in add_out_dense_sparse_compressed_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distance.cpp | 305 auto commonDtype = at::result_type(x1_, x2_); in cosine_similarity() local
|