Home
last modified time | relevance | path

Searched refs:computeDispatchKey (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DTensorOptions.h26 DispatchKey computeDispatchKey(
134 DispatchKey computeDispatchKey(
376 return computeDispatchKey() == other.computeDispatchKey() && in type_equal()
405 return at::dispatchKeyToBackend(computeDispatchKey()); in backend()
450 DispatchKey computeDispatchKey() const { in computeDispatchKey() function
451 return c10::computeDispatchKey( in computeDispatchKey()
634 inline DispatchKey computeDispatchKey( in computeDispatchKey() function
/aosp_15_r20/external/pytorch/aten/src/ATen/quantized/
H A DQuantizer.cpp138 at::DispatchKey tensorDispatchKey = options.computeDispatchKey(); in new_qtensor()
352 at::DispatchKeySet(options.computeDispatchKey()), in from_blob_quantized_per_tensor_affine()
427 at::DispatchKeySet(options.computeDispatchKey()), in from_blob_quantized_per_channel_affine()
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DFunctions.cpp44 std::move(storage), opts_.computeDispatchKey(), opts_.dtype()); in make_tensor()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensor.cpp86 auto dispatchkey = options().computeDispatchKey(); in toString()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp930 c10::computeDispatchKey(std::nullopt, std::nullopt, device)); in initDispatchBindings()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp848 auto keys = c10::DispatchKeySet({options.computeDispatchKey()}); in THPVariable_make_wrapper_subclass()