Searched defs:dispatchKeySet (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/ |
| H A D | KernelFunction_impl.h | 45 inline void KernelFunction::callBoxed(const OperatorHandle& opHandle, DispatchKeySet dispatchKeySet… in callBoxed()
50 …lFunction(void* unboxed_kernel_func, OperatorKernel* functor, DispatchKeySet dispatchKeySet, Args&… in callUnboxedKernelFunction()
84 …E Return KernelFunction::call(const OperatorHandle& opHandle, DispatchKeySet dispatchKeySet, Args.… in call()
|
| H A D | BoxedKernel_impl.h | 36 inline void BoxedKernel::callBoxed(const OperatorHandle& opHandle, DispatchKeySet dispatchKeySet, S… in callBoxed()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | Dispatcher.cpp | 57 …trace(const std::string& label, const std::string& op_name, const DispatchKeySet& dispatchKeySet) { in _print_dispatch_trace()
508 …r::sequenceNumberForRunningRecordFunction(DispatchKey dispatchKey, DispatchKeySet dispatchKeySet) { in sequenceNumberForRunningRecordFunction()
532 …rdFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, DispatchKeySet dispatchKeySet, c10::… in runRecordFunction()
536 …:RecordFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, DispatchKeySet dispatchKeySet) { in runRecordFunction()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | FunctionalizeFallbackKernel.cpp | 31 …void functionalizeFallback(const c10::OperatorHandle& op, c10::DispatchKeySet dispatchKeySet [[may… in functionalizeFallback()
128 static const at::Tensor & resize__functionalization(c10::DispatchKeySet dispatchKeySet [[maybe_unus… in resize__functionalization()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | VariableTypeUtils.h | 417 c10::DispatchKeySet dispatchKeySet, in run_jit_decomposition_with_args_for_jvp()
|