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: