Home
last modified time | relevance | path

Searched refs:highestPriorityBackendTypeId (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/c10/test/core/
DDispatchKeySet_test.cpp323 ASSERT_EQ(DispatchKey::CPU, c10::highestPriorityBackendTypeId(dense_cpu)); in TEST()
329 DispatchKey::SparseCUDA, c10::highestPriorityBackendTypeId(sparse_cuda)); in TEST()
335 c10::highestPriorityBackendTypeId(sparse_compressed_cuda)); in TEST()
342 c10::highestPriorityBackendTypeId(quantized_cuda)); in TEST()
/external/pytorch/aten/src/ATen/
DSparseTensorImpl.cpp10 auto k = c10::highestPriorityBackendTypeId(key_set); in sparseTensorSetToDeviceType()
/external/pytorch/c10/core/
DDispatchKeySet.h894 inline DispatchKey highestPriorityBackendTypeId(DispatchKeySet ks) { in highestPriorityBackendTypeId() function