Home
last modified time | relevance | path

Searched refs:dispatchKeyToBackend (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp727 if (isSparse(dispatchKeyToBackend(dispatch_key))) { in legacy_tensor_generic_ctor_new()
891 TORCH_INTERNAL_ASSERT(!isSparseCsr(dispatchKeyToBackend(dispatch_key))); in sparse_compressed_tensor_ctor_worker()
892 TORCH_INTERNAL_ASSERT(!isSparse(dispatchKeyToBackend(dispatch_key))); in sparse_compressed_tensor_ctor_worker()
1137 TORCH_INTERNAL_ASSERT(!isSparse(dispatchKeyToBackend(dispatch_key))); in sparse_coo_tensor_ctor()
1138 TORCH_INTERNAL_ASSERT(!isSparseCsr(dispatchKeyToBackend(dispatch_key))); in sparse_coo_tensor_ctor()
H A Dtensor_types.cpp102 dispatchKeyToBackend(torch::tensors::get_default_dispatch_key()); in options_from_string()
/aosp_15_r20/external/pytorch/c10/core/
H A DBackend.h68 inline Backend dispatchKeyToBackend(DispatchKey t) { in dispatchKeyToBackend() function
H A DTensorOptions.h405 return at::dispatchKeyToBackend(computeDispatchKey()); in backend()
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DTensorBody.h227 dispatchKeyToBackend(legacyExtractDispatchKey(key_set())), in type()