Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::CudaRTCFunction< Derived > Class Template Reference

Public Member Functions

template<typename... Args>
void Compile (Args...args)
 
template<typename... Args>
void Launch (unsigned int gx, unsigned int gy, unsigned int gz, unsigned int bx, unsigned int by, unsigned int bz, unsigned int shared_mem, cudaStream_t stream, Args...args)
 
void LaunchEx (unsigned int gx, unsigned int gy, unsigned int gz, unsigned int bx, unsigned int by, unsigned int bz, unsigned int shared_mem, cudaStream_t stream, void **extra)
 

Detailed Description

template<typename Derived>
class caffe2::CudaRTCFunction< Derived >

Definition at line 22 of file common_rtc.h.


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