mxnet
mxnet::KVStore Member List

This is the complete list of members for mxnet::KVStore, including all inherited members.

Barrier()mxnet::KVStoreinlinevirtual
barrier_before_exit_mxnet::KVStoreprotected
Broadcast(const std::vector< int > &vkeys, const std::vector< int > &okeys, const std::vector< NDArray > &values, const std::vector< NDArray * > &outs, int priority=0)=0mxnet::KVStorepure virtual
Broadcast(const std::vector< std::string > &str_vkeys, const std::vector< std::string > &str_okeys, const std::vector< NDArray > &values, const std::vector< NDArray * > &outs, int priority=0)=0mxnet::KVStorepure virtual
Controller typedefmxnet::KVStore
Create(const char *type="local")mxnet::KVStorestatic
get_group_size() const mxnet::KVStoreinlinevirtual
get_num_dead_node(int node_id, int timeout=60) const mxnet::KVStoreinlinevirtual
get_rank() const mxnet::KVStoreinlinevirtual
gradient_compression_mxnet::KVStoreprotected
Init(const std::vector< int > &keys, const std::vector< NDArray > &values)=0mxnet::KVStorepure virtual
Init(const std::vector< std::string > &str_keys, const std::vector< NDArray > &values)=0mxnet::KVStorepure virtual
InitPSEnv(const std::unordered_map< std::string, std::string > &envs)mxnet::KVStoreinlinestatic
IsSchedulerNode()mxnet::KVStoreinlinestatic
IsServerNode()mxnet::KVStoreinlinestatic
IsWorkerNode()mxnet::KVStoreinlinestatic
Pull(const std::vector< int > &keys, const std::vector< NDArray * > &values, int priority=0, bool ignore_sparse=true)=0mxnet::KVStorepure virtual
Pull(const std::vector< std::string > &str_keys, const std::vector< NDArray * > &values, int priority=0, bool ignore_sparse=true)=0mxnet::KVStorepure virtual
PullRowSparse(const std::vector< int > &str_keys, const std::vector< std::pair< NDArray *, NDArray >> &val_rowids, int priority=0)=0mxnet::KVStorepure virtual
PullRowSparse(const std::vector< std::string > &str_keys, const std::vector< std::pair< NDArray *, NDArray >> &val_rowids, int priority=0)=0mxnet::KVStorepure virtual
Push(const std::vector< int > &keys, const std::vector< NDArray > &values, int priority=0)=0mxnet::KVStorepure virtual
Push(const std::vector< std::string > &str_keys, const std::vector< NDArray > &values, int priority=0)=0mxnet::KVStorepure virtual
PushPull(const std::vector< int > &vkeys, const std::vector< int > &okeys, const std::vector< NDArray > &values, const std::vector< NDArray * > &outs, int priority=0)=0mxnet::KVStorepure virtual
PushPull(const std::vector< std::string > &str_vkeys, const std::vector< std::string > &str_okeys, const std::vector< NDArray > &values, const std::vector< NDArray * > &outs, int priority=0)=0mxnet::KVStorepure virtual
RunServer(const Controller &controller)mxnet::KVStoreinlinevirtual
SendCommandToServers(int cmd_id, const std::string &cmd_body)mxnet::KVStoreinlinevirtual
set_barrier_before_exit(const bool barrier_before_exit)mxnet::KVStoreinline
set_updater(const Updater &updater)mxnet::KVStoreinlinevirtual
set_updater(const StrUpdater &updater)mxnet::KVStoreinlinevirtual
SetGradientCompression(const std::vector< std::pair< std::string, std::string > > &kwargs)=0mxnet::KVStorepure virtual
SetServerProfilerCommand(const KVStoreServerProfilerCommand type, const std::string &params)mxnet::KVStoreinlinevirtual
str_updater_mxnet::KVStoreprotected
StrUpdater typedefmxnet::KVStore
type()mxnet::KVStoreinline
type_mxnet::KVStoreprotected
Updater typedefmxnet::KVStore
updater_mxnet::KVStoreprotected
~KVStore()mxnet::KVStoreinlinevirtual