Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Enumerator Macros
Public Types | Public Member Functions | List of all members
SyncedMemory Class Reference

Manages memory allocation and synchronization between the host (CPU) and device (GPU). More...

#include <blob.h>

Public Types

enum  SyncedHead { UNINITIALIZED, HEAD_AT_CPU, HEAD_AT_GPU, SYNCED }
 

Public Member Functions

 SyncedMemory (size_t size)
 
const void * cpu_data ()
 
void set_cpu_data (void *data)
 
const void * gpu_data ()
 
void * mutable_cpu_data ()
 
void * mutable_gpu_data ()
 
SyncedHead head ()
 
size_t size ()
 

Detailed Description

Manages memory allocation and synchronization between the host (CPU) and device (GPU).

TODO(dox): more thorough description.


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