Searched refs:currentDispatchKeySet (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/core/dispatch/ |
| D | Dispatcher.h | 181 …h(const TypedOperatorHandle<Return (Args...)>& op, DispatchKeySet currentDispatchKeySet, Args... a…
535 C10_ALWAYS_INLINE Return redispatch(DispatchKeySet currentDispatchKeySet, Args... args) const { in redispatch() argument
536 …return c10::Dispatcher::singleton().redispatch<Return, Args...>(*this, currentDispatchKeySet, std:… in redispatch()
704 …h(const TypedOperatorHandle<Return (Args...)>& op, DispatchKeySet currentDispatchKeySet, Args... a…
710 …detail::_print_dispatch_trace("[redispatch]", toString(op.operator_name()), currentDispatchKeySet);
713 const KernelFunction& kernel = op.operatorDef_->op.lookup(currentDispatchKeySet);
714 …return kernel.template call<Return, Args...>(op, currentDispatchKeySet, std::forward<Args>(args)..…
|