Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Static Public Member Functions | Protected Attributes
caffe2::IDEEPContext Class Referencefinal
Inheritance diagram for caffe2::IDEEPContext:
at::BaseContext

Public Types

typedef std::mt19937 rand_gen_type
 

Public Member Functions

 IDEEPContext (const DeviceOption &option)
 
 IDEEPContext (const at::Device &device)
 
void SwitchToDevice (int)
 
void WaitEvent (const Event &ev)
 
void Record (Event *ev, const char *err_msg=nullptr) const
 
void FinishDeviceComputation ()
 
rand_gen_type & RandGenerator ()
 
void CopyBytesSameDevice (size_t nbytes, const void *src, void *dst) override
 
void CopyBytesFromCPU (size_t nbytes, const void *src, void *dst) override
 
void CopyBytesToCPU (size_t nbytes, const void *src, void *dst) override
 
bool SupportsNonFundamentalTypes () const override
 
template<class SrcContext , class DstContext >
void CopyBytes (size_t nbytes, const void *src, void *dst)
 
template<typename T , class SrcContext , class DstContext >
void Copy (size_t n, const T *src, T *dst)
 
template<class SrcContext , class DstContext >
void CopyItems (const TypeMeta &meta, size_t n, const void *src, void *dst)
 
at::Device device () const override
 
DeviceType device_type () const override
 
template<>
void CopyBytes (size_t nbytes, const void *src, void *dst)
 
template<>
void CopyBytes (size_t nbytes, const void *src, void *dst)
 
template<>
void CopyBytes (size_t nbytes, const void *src, void *dst)
 
- Public Member Functions inherited from at::BaseContext
void SwitchToDevice ()
 
template<typename T >
void CopySameDevice (size_t n, const T *src, T *dst)
 
template<typename T >
void CopyFromCPU (size_t n, const T *src, T *dst)
 
template<typename T >
void CopyToCPU (size_t n, const T *src, T *dst)
 
void EnforceMetaCopyOK ()
 
void CopyItemsSameDevice (const caffe2::TypeMeta &meta, size_t n, const void *src, void *dst)
 
void CopyItemsFromCPU (const caffe2::TypeMeta &meta, size_t n, const void *src, void *dst)
 
void CopyItemsToCPU (const caffe2::TypeMeta &meta, size_t n, const void *src, void *dst)
 

Static Public Member Functions

static at::DataPtr New (size_t nbytes)
 
static bool HasAsyncPartDefault ()
 
static bool SupportsAsyncScheduling ()
 
static bool IsStreamFree (const DeviceOption &, int)
 
static constexpr DeviceType GetDeviceType ()
 

Protected Attributes

int random_seed_ {1701}
 
std::unique_ptr< rand_gen_type > random_generator_
 

Detailed Description

Definition at line 11 of file ideep_context.h.


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