mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
7663ffb673
Still a continued work for clean up c10/util/*.h Pull Request resolved: https://github.com/pytorch/pytorch/pull/116326 Approved by: https://github.com/Skylion007
32 lines
727 B
C++
32 lines
727 B
C++
#pragma once
|
|
|
|
#include <memory>
|
|
namespace c10 {
|
|
namespace detail {
|
|
|
|
template <class... Ts>
|
|
struct overloaded_t {};
|
|
|
|
template <class T0>
|
|
struct overloaded_t<T0> : T0 {
|
|
using T0::operator();
|
|
overloaded_t(T0 t0) : T0(std::move(t0)) {}
|
|
};
|
|
template <class T0, class... Ts>
|
|
struct overloaded_t<T0, Ts...> : T0, overloaded_t<Ts...> {
|
|
using T0::operator();
|
|
using overloaded_t<Ts...>::operator();
|
|
overloaded_t(T0 t0, Ts... ts)
|
|
: T0(std::move(t0)), overloaded_t<Ts...>(std::move(ts)...) {}
|
|
};
|
|
|
|
} // namespace detail
|
|
|
|
// Construct an overloaded callable combining multiple callables, e.g. lambdas
|
|
template <class... Ts>
|
|
detail::overloaded_t<Ts...> overloaded(Ts... ts) {
|
|
return {std::move(ts)...};
|
|
}
|
|
|
|
} // namespace c10
|