Searched defs:MultiDispatchKeySet (Results 1 – 1 of 1) sorted by relevance
54 struct MultiDispatchKeySet : at::IterArgs<MultiDispatchKeySet> { struct55 DispatchKeySet ts;56 void operator()(const at::Tensor& x) { in operator()59 void operator()(const std::optional<at::Tensor>& x) { in operator()64 void operator()(at::ArrayRef<at::Tensor> xs) { in operator()70 void operator()(const c10::List<std::optional<at::Tensor>>& xs) { in operator()78 void operator()(const at::ITensorListRef& xs) { in operator()83 [[noreturn]] void operator()(at::ArrayRef<std::optional<at::Tensor>>) { in operator()107 return MultiDispatchKeySet().apply(args...).ts; in multi_dispatch_key_set() argument