#include <kvstore.h>

Collaboration diagram for mxnet::cpp::KVStore:

Static Public Member Functions
static void	SetType (const std::string &type)

static void	RunServer ()

static void	Init (int key, const NDArray &val)

static void	Init (const std::string &key, const NDArray &val)

static void	Init (const std::vector< int > &keys, const std::vector< NDArray > &vals)

static void	Init (const std::vector< std::string > &keys, const std::vector< NDArray > &vals)

static void	Push (int key, const NDArray &val, int priority=0)

static void	Push (const std::string &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	Push (const std::vector< std::string > &keys, const std::vector< NDArray > &vals, int priority=0)

static void	Pull (int key, NDArray *out, int priority=0)

static void	Pull (const std::string &key, NDArray *out, int priority=0)

static void	Pull (const std::vector< int > &keys, std::vector< NDArray > *outs, int priority=0)

static void	Pull (const std::vector< std::string > &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 ()

Member Function Documentation

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::string &	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::Init	(	const std::vector< std::string > &	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::string &	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::Pull	(	const std::vector< std::string > &	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::string &	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::Push	(	const std::vector< std::string > &	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:

Static Public Member Functions