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,
49 static void Push(
const std::vector<std::string>& keys,
50 const std::vector<NDArray>& vals,
52 static void Pull(
int key,
NDArray* out,
int priority = 0);
53 static void Pull(
const std::string& key,
NDArray* out,
int priority = 0);
54 static void Pull(
const std::vector<int>& keys, std::vector<NDArray>* outs,
int priority = 0);
55 static void Pull(
const std::vector<std::string>& keys,
56 std::vector<NDArray>* outs,
59 static void SetOptimizer(std::unique_ptr<Optimizer> optimizer,
bool local =
false);
69 static std::unique_ptr<Optimizer>& get_optimizer();
71 static void Controller(
int head,
const char* body,
void* controller_handle);
78 #endif // MXNET_CPP_KVSTORE_H_