Home
last modified time | relevance | path

Searched full:sparsecsrcuda (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DBackend.h40 SparseCsrCUDA, enumerator
105 } else if (t == DispatchKey::SparseCsrCUDA) { in dispatchKeyToBackend()
106 return Backend::SparseCsrCUDA; in dispatchKeyToBackend()
183 case Backend::SparseCsrCUDA: in backendToDispatchKey()
184 return DispatchKey::SparseCsrCUDA; in backendToDispatchKey()
231 case Backend::SparseCsrCUDA: in backendToDeviceType()
322 case Backend::SparseCsrCUDA: in toString()
323 return "SparseCsrCUDA"; in toString()
377 case Backend::SparseCsrCUDA: in isSparseCsr()
H A DDispatchKey.cpp351 {"SparseCsrCUDA", c10::DispatchKey::SparseCsrCUDA}, in parseDispatchKey()
H A DLayout.h43 case Backend::SparseCsrCUDA: in layout_from_backend()
H A DDispatchKeySet.h82 // - "SparseCsr": SparseCsrCPU, SparseCsrCUDA, ...
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml341 SparseCsrCPU, SparseCsrCUDA: abs_sparse_csr
351 SparseCsrCPU, SparseCsrCUDA: abs_sparse_csr_
360 SparseCsrCPU, SparseCsrCUDA: abs_sparse_csr_out
403 SparseCsrCPU, SparseCsrCUDA: angle_sparse_csr
410 SparseCsrCPU, SparseCsrCUDA: angle_sparse_csr_out
428 SparseCsrCPU, SparseCsrCUDA: sgn_sparse_csr
437 SparseCsrCPU, SparseCsrCUDA: sgn_sparse_csr_
448 SparseCsrCPU, SparseCsrCUDA: sgn_sparse_csr_out
475 SparseCsrCPU, SparseCsrCUDA: conj_physical_sparse_csr
487 SparseCsrCPU, SparseCsrCUDA: conj_physical_sparse_csr_out
[all …]
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_executorch_gen.py40 SparseCsrCUDA: add_out_sparse_csr_cuda
50 SparseCsrCPU, SparseCsrCUDA: add_sparse_csr
65 SparseCsrCPU, SparseCsrCUDA: mul_out_sparse_csr
74 SparseCsrCPU, SparseCsrCUDA: mul_sparse_csr
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorImpl.cpp58 || (key_set.has(DispatchKey::SparseCsrCUDA) && device().type() == kCUDA) in SparseCsrTensorImpl()
61 … || (key_set.has(DispatchKey::SparseCsrCUDA) && device().type() == kMeta) // fake tensor in SparseCsrTensorImpl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseBlas.cpp103 // when using the same function for CUDA and SparseCsrCUDA dispatch keys in addmm_out_sparse_compressed_cuda()
227 // problems, when using the same function for CUDA and SparseCsrCUDA dispatch in addmv_out_sparse_compressed_cuda()
H A DSparseCsrTensorMath.cu403 SparseCsrCUDA: _sparse_csr_..._cuda
/aosp_15_r20/external/pytorch/c10/test/core/
H A DDispatchKeySet_test.cpp332 {DispatchKey::Functionalize, DispatchKey::SparseCsrCUDA}); in TEST()
334 DispatchKey::SparseCsrCUDA, in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseBlasImpl.cpp115 if (triton_kernel.hasKernelForDispatchKey(c10::DispatchKey::SparseCsrCUDA)) { in _compressed_row_strided_mm_out()
244 if (triton_kernel.hasKernelForDispatchKey(c10::DispatchKey::SparseCsrCUDA)) { in _compressed_row_strided_addmm_out()
H A DSparseCsrTensor.cpp345 // SparseCsrCPU, SparseCsrCUDA, and SparseCsrTensorImpl exists because
361 dispatch_key = DispatchKey::SparseCsrCUDA; in new_compressed_tensor()
H A DSparseCsrTensorMath.cpp1006 SparseCsrCUDA: _sparse_csr_..._cpu
/aosp_15_r20/external/pytorch/torch/cuda/
H A D__init__.py1525 "SparseCsrCUDA",
1535 "SparseCsrCUDA",
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py181 SparseCsrCUDA = auto() variable in DispatchKey
283 DispatchKey.SparseCsrCUDA,
320 DispatchKey.SparseCsrCUDA,
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py3621 "SparseCsrCUDA"