Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Data Fields | Static Public Attributes
at::native::DispatchStub< rT(*)(Args...), T > Struct Template Reference

Public Types

using FnPtr = rT(*)(Args...)

Public Member Functions

template<typename... ArgTypes>
rT operator() (DeviceType device_type, ArgTypes &&...args)
FnPtr choose_cpu_impl ()

Data Fields

FnPtr cpu_dispatch_ptr = nullptr
FnPtr cuda_dispatch_ptr = nullptr
FnPtr hip_dispatch_ptr = nullptr

Static Public Attributes

static FnPtr DEFAULT

Detailed Description

template<typename rT, typename T, typename... Args>
struct at::native::DispatchStub< rT(*)(Args...), T >

Definition at line 59 of file DispatchStub.h.

The documentation for this struct was generated from the following file: