#include <kvstore.h>
|
| static void | SetType (const std::string &type) |
| |
| static void | RunServer () |
| |
| static void | Init (int key, const NDArray &val) |
| |
| static void | Init (const std::vector< int > &keys, const std::vector< NDArray > &vals) |
| |
| static void | Push (int 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 | Pull (int key, NDArray *out, int priority=0) |
| |
| static void | Pull (const std::vector< int > &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 () |
| |
| 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::vector< int > & |
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::vector< int > & |
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::vector< int > & |
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:
1.8.11
