mxnet
|
This is the complete list of members for mxnet::KVStore, including all inherited members.
Barrier() | mxnet::KVStore | inlinevirtual |
barrier_before_exit_ | mxnet::KVStore | protected |
Controller typedef | mxnet::KVStore | |
Create(const char *type="local") | mxnet::KVStore | static |
get_group_size() const | mxnet::KVStore | inlinevirtual |
get_num_dead_node(int node_id, int timeout=60) const | mxnet::KVStore | inlinevirtual |
get_rank() const | mxnet::KVStore | inlinevirtual |
gradient_compression_ | mxnet::KVStore | protected |
Init(const std::vector< int > &keys, const std::vector< NDArray > &values)=0 | mxnet::KVStore | pure virtual |
Init(const std::vector< std::string > &str_keys, const std::vector< NDArray > &values)=0 | mxnet::KVStore | pure virtual |
InitPSEnv(const std::unordered_map< std::string, std::string > &envs) | mxnet::KVStore | inlinestatic |
IsSchedulerNode() | mxnet::KVStore | inlinestatic |
IsServerNode() | mxnet::KVStore | inlinestatic |
IsWorkerNode() | mxnet::KVStore | inlinestatic |
Pull(const std::vector< int > &keys, const std::vector< NDArray * > &values, int priority=0)=0 | mxnet::KVStore | pure virtual |
Pull(const std::vector< std::string > &str_keys, const std::vector< NDArray * > &values, int priority=0)=0 | mxnet::KVStore | pure virtual |
PullRowSparse(const std::vector< int > &str_keys, const std::vector< std::pair< NDArray *, NDArray >> &val_rowids, int priority=0)=0 | mxnet::KVStore | pure virtual |
PullRowSparse(const std::vector< std::string > &str_keys, const std::vector< std::pair< NDArray *, NDArray >> &val_rowids, int priority=0)=0 | mxnet::KVStore | pure virtual |
Push(const std::vector< int > &keys, const std::vector< NDArray > &values, int priority=0)=0 | mxnet::KVStore | pure virtual |
Push(const std::vector< std::string > &str_keys, const std::vector< NDArray > &values, int priority=0)=0 | mxnet::KVStore | pure virtual |
RunServer(const Controller &controller) | mxnet::KVStore | inlinevirtual |
SendCommandToServers(int cmd_id, const std::string &cmd_body) | mxnet::KVStore | inlinevirtual |
set_barrier_before_exit(const bool barrier_before_exit) | mxnet::KVStore | inline |
set_updater(const Updater &updater) | mxnet::KVStore | inlinevirtual |
set_updater(const StrUpdater &updater) | mxnet::KVStore | inlinevirtual |
SetGradientCompression(const std::vector< std::pair< std::string, std::string > > &kwargs)=0 | mxnet::KVStore | pure virtual |
str_updater_ | mxnet::KVStore | protected |
StrUpdater typedef | mxnet::KVStore | |
type() | mxnet::KVStore | inline |
type_ | mxnet::KVStore | protected |
Updater typedef | mxnet::KVStore | |
updater_ | mxnet::KVStore | protected |
~KVStore() | mxnet::KVStore | inlinevirtual |