26 #ifndef MXNET_CPP_KVSTORE_H_ 27 #define MXNET_CPP_KVSTORE_H_ 38 static void SetType(
const std::string& type);
41 static void Init(
const std::string& key,
const NDArray& val);
42 static void Init(
const std::vector<int>& keys,
const std::vector<NDArray>& vals);
43 static void Init(
const std::vector<std::string>& keys,
const std::vector<NDArray>& vals);
44 static void Push(
int key,
const NDArray& val,
int priority = 0);
45 static void Push(
const std::string& key,
const NDArray& val,
int priority = 0);
46 static void Push(
const std::vector<int>& keys,
47 const std::vector<NDArray>& vals,
int priority = 0);
48 static void Push(
const std::vector<std::string>& keys,
49 const std::vector<NDArray>& vals,
int priority = 0);
50 static void Pull(
int key,
NDArray* out,
int priority = 0);
51 static void Pull(
const std::string& key,
NDArray* out,
int priority = 0);
52 static void Pull(
const std::vector<int>& keys,
53 std::vector<NDArray>* outs,
int priority = 0);
54 static void Pull(
const std::vector<std::string>& keys,
55 std::vector<NDArray>* outs,
int priority = 0);
57 static void SetOptimizer(std::unique_ptr<Optimizer> optimizer,
bool local =
false);
67 static std::unique_ptr<Optimizer>& get_optimizer();
69 static void Controller(
int head,
const char* body,
void* controller_handle);
76 #endif // MXNET_CPP_KVSTORE_H_ static std::string GetRole()
static void Pull(int key, NDArray *out, int priority=0)
static void Init(int key, const NDArray &val)
static void SetType(const std::string &type)
namespace of mxnet
Definition: api_registry.h:33
static void SetOptimizer(std::unique_ptr< Optimizer > optimizer, bool local=false)
NDArray interface.
Definition: ndarray.h:120
static void Push(int key, const NDArray &val, int priority=0)
static int GetNumWorkers()
static std::string GetType()
void * NDArrayHandle
handle to NDArray
Definition: c_api.h:66
void * KVStoreHandle
handle to KVStore
Definition: c_api.h:84