mxnet
Static Public Member Functions | List of all members
mxnet::cpp::KVStore Class Reference

#include <kvstore.h>

Collaboration diagram for mxnet::cpp::KVStore:
Collaboration graph

Static Public Member Functions

static void SetType (const std::string &type)
 
static void RunServer ()
 
static void Init (int key, const NDArray &val)
 
static void Init (const std::string &key, const NDArray &val)
 
static void Init (const std::vector< int > &keys, const std::vector< NDArray > &vals)
 
static void Init (const std::vector< std::string > &keys, const std::vector< NDArray > &vals)
 
static void Push (int key, const NDArray &val, int priority=0)
 
static void Push (const std::string &key, const NDArray &val, int priority=0)
 
static void Push (const std::vector< int > &keys, const std::vector< NDArray > &vals, int priority=0)
 
static void Push (const std::vector< std::string > &keys, const std::vector< NDArray > &vals, int priority=0)
 
static void Pull (int key, NDArray *out, int priority=0)
 
static void Pull (const std::string &key, NDArray *out, int priority=0)
 
static void Pull (const std::vector< int > &keys, std::vector< NDArray > *outs, int priority=0)
 
static void Pull (const std::vector< std::string > &keys, std::vector< NDArray > *outs, int priority=0)
 
static void SetOptimizer (std::unique_ptr< Optimizer > optimizer, bool local=false)
 
static std::string GetType ()
 
static int GetRank ()
 
static int GetNumWorkers ()
 
static void Barrier ()
 
static std::string GetRole ()
 

Member Function Documentation

static void mxnet::cpp::KVStore::Barrier ( )
static
static int mxnet::cpp::KVStore::GetNumWorkers ( )
static
static int mxnet::cpp::KVStore::GetRank ( )
static
static std::string mxnet::cpp::KVStore::GetRole ( )
static
static std::string mxnet::cpp::KVStore::GetType ( )
static
static void mxnet::cpp::KVStore::Init ( int  key,
const NDArray val 
)
static
static void mxnet::cpp::KVStore::Init ( const std::string &  key,
const NDArray val 
)
static
static void mxnet::cpp::KVStore::Init ( const std::vector< int > &  keys,
const std::vector< NDArray > &  vals 
)
static
static void mxnet::cpp::KVStore::Init ( const std::vector< std::string > &  keys,
const std::vector< NDArray > &  vals 
)
static
static void mxnet::cpp::KVStore::Pull ( int  key,
NDArray out,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Pull ( const std::string &  key,
NDArray out,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Pull ( const std::vector< int > &  keys,
std::vector< NDArray > *  outs,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Pull ( const std::vector< std::string > &  keys,
std::vector< NDArray > *  outs,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Push ( int  key,
const NDArray val,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Push ( const std::string &  key,
const NDArray val,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Push ( const std::vector< int > &  keys,
const std::vector< NDArray > &  vals,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::Push ( const std::vector< std::string > &  keys,
const std::vector< NDArray > &  vals,
int  priority = 0 
)
static
static void mxnet::cpp::KVStore::RunServer ( )
static
static void mxnet::cpp::KVStore::SetOptimizer ( std::unique_ptr< Optimizer optimizer,
bool  local = false 
)
static
static void mxnet::cpp::KVStore::SetType ( const std::string &  type)
static

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