Searched defs:dispatchKey (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/core/dispatch/ |
| D | Dispatcher.cpp | 421 RegistrationHandleRAII Dispatcher::registerFallback(DispatchKey dispatchKey, KernelFunction kernel,… in registerFallback()
448 void Dispatcher::deregisterFallback_(DispatchKey dispatchKey) { in deregisterFallback_()
508 int64_t Dispatcher::sequenceNumberForRunningRecordFunction(DispatchKey dispatchKey, DispatchKeySet … in sequenceNumberForRunningRecordFunction()
532 …ordFunction& guard, at::RecordFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, Dispatch… in runRecordFunction()
536 …ordFunction& guard, at::RecordFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, Dispatch… in runRecordFunction()
|
| /external/pytorch/aten/src/ATen/core/boxing/impl/ |
| D | test_helpers.h | 55 …t callOpUnboxedWithDispatchKey(const c10::OperatorHandle& op, c10::DispatchKey dispatchKey, Args..… in callOpUnboxedWithDispatchKey()
|