Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/core/dispatch/
DDispatcher.cpp421 RegistrationHandleRAII Dispatcher::registerFallback(DispatchKey dispatchKey, KernelFunction kernel,… in registerFallback() argument
424 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in registerFallback()
428 …"Tried to register multiple backend fallbacks for the same dispatch key ", dispatchKey, "; previou… in registerFallback()
436 op.op.updateFallback(*this, dispatchKey); in registerFallback()
439 return RegistrationHandleRAII([guard = this->guard_, this, dispatchKey] { in registerFallback()
444 deregisterFallback_(dispatchKey); in registerFallback()
448 void Dispatcher::deregisterFallback_(DispatchKey dispatchKey) { in deregisterFallback_() argument
449 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in deregisterFallback_()
453 op.op.updateFallback(*this, dispatchKey); in deregisterFallback_()
508 int64_t Dispatcher::sequenceNumberForRunningRecordFunction(DispatchKey dispatchKey, DispatchKeySet … in sequenceNumberForRunningRecordFunction() argument
[all …]
DDispatcher.h307 …static int64_t sequenceNumberForRunningRecordFunction(DispatchKey dispatchKey, DispatchKeySet disp…
308 …ction& guard, at::RecordFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, DispatchKeySet…
309 …ction& guard, at::RecordFunction::schema_ref_t schema_ref, DispatchKey dispatchKey, DispatchKeySet…
327 void deregisterFallback_(DispatchKey dispatchKey);
621 auto dispatchKey = dispatchKeySet.highestPriorityTypeId();
639 …runRecordFunction(guard, schema_ref, dispatchKey, dispatchKeySet, c10::ArrayRef<const c10::IValue>…
644 runRecordFunction(guard, schema_ref, dispatchKey, dispatchKeySet);
647 runRecordFunction(guard, schema_ref, dispatchKey, dispatchKeySet);
732 auto dispatchKey = dispatchKeySet.highestPriorityTypeId();
735 …guard.needsInputs() ? runRecordFunction(guard, schema_ref, dispatchKey, dispatchKeySet, c10::Array…
[all …]
DOperatorEntry.cpp533 void OperatorEntry::reportError(DispatchKey dispatchKey) const { in registerKernel()
538 …tErrorCallback(report_error_callback_->ptr(&report_error_callback_->pyinterpreter()), dispatchKey); in registerKernel()
542 if (dispatchKey == DispatchKey::Undefined) { in registerKernel()
552 …" from the '", post_process_dispatch_key_str(toString(dispatchKey)), "' backend. This could be bec… in registerKernel()
DOperatorEntry.h172 [[noreturn]] void reportError(DispatchKey dispatchKey) const;
/external/pytorch/aten/src/ATen/core/boxing/impl/
Dtest_helpers.h55 …UnboxedWithDispatchKey(const c10::OperatorHandle& op, c10::DispatchKey dispatchKey, Args... args) { in callOpUnboxedWithDispatchKey() argument
56 return op.typed<Result(Args...)>().callWithDispatchKey(dispatchKey, std::forward<Args>(args)...); in callOpUnboxedWithDispatchKey()