MLCommon::Stats::opg Namespace Reference

Functions
void	cov (const raft::handle_t &handle, Matrix::Data< float > &covar, const std::vector< Matrix::Data< float > * > &data, const Matrix::PartDescriptor &dataDesc, Matrix::Data< float > &mu, bool sample, cudaStream_t *streams, int n_streams)
	performs MNMG covariance calculation. More...
void	cov (const raft::handle_t &handle, Matrix::Data< double > &covar, const std::vector< Matrix::Data< double > * > &data, const Matrix::PartDescriptor &dataDesc, Matrix::Data< double > &mu, bool sample, cudaStream_t *streams, int n_streams)
void	mean (const raft::handle_t &handle, Matrix::Data< float > &out, const std::vector< Matrix::Data< float > * > &in, const Matrix::PartDescriptor &inDesc, cudaStream_t *streams, int n_streams)
	performs MNMG mean calculation. More...
void	mean (const raft::handle_t &handle, Matrix::Data< double > &out, const std::vector< Matrix::Data< double > * > &in, const Matrix::PartDescriptor &inDesc, cudaStream_t *streams, int n_streams)
void	mean_center (const std::vector< Matrix::Data< double > * > &data, const Matrix::PartDescriptor &dataDesc, const Matrix::Data< double > &mu, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
	performs MNMG mean subtraction calculation. More...
void	mean_center (const std::vector< Matrix::Data< float > * > &data, const Matrix::PartDescriptor &dataDesc, const Matrix::Data< float > &mu, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
void	mean_add (const std::vector< Matrix::Data< double > * > &data, const Matrix::PartDescriptor &dataDesc, const Matrix::Data< double > &mu, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
	performs MNMG mean add calculation. More...
void	mean_add (const std::vector< Matrix::Data< float > * > &data, const Matrix::PartDescriptor &dataDesc, const Matrix::Data< float > &mu, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
void	var (const raft::handle_t &handle, Matrix::Data< float > &out, const std::vector< Matrix::Data< float > * > &in, const Matrix::PartDescriptor &inDesc, const float *mu, cudaStream_t *streams, int n_streams)
	performs MNMG var calculation. More...
void	var (const raft::handle_t &handle, Matrix::Data< double > &out, const std::vector< Matrix::Data< double > * > &in, const Matrix::PartDescriptor &inDesc, const double *mu, cudaStream_t *streams, int n_streams)

Function Documentation

cov() [1/2]

void MLCommon::Stats::opg::cov	(	const raft::handle_t &	handle,
		Matrix::Data< double > &	covar,
		const std::vector< Matrix::Data< double > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		Matrix::Data< double > &	mu,
		bool	sample,
		cudaStream_t *	streams,
		int	n_streams
	)

cov() [2/2]

void MLCommon::Stats::opg::cov	(	const raft::handle_t &	handle,
		Matrix::Data< float > &	covar,
		const std::vector< Matrix::Data< float > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		Matrix::Data< float > &	mu,
		bool	sample,
		cudaStream_t *	streams,
		int	n_streams
	)

performs MNMG covariance calculation.

Parameters

[in]	handle	cuML handle object
[out]	covar	resulting covariance matrix
[in]	data	the data that cov matrix is calculated for
[in]	dataDesc	MNMG description of the input data
[in]	mu	mean of every column in data
[in]	sample	whether to compute sample covariance
[in]	streams	cuda streams
[in]	n_streams	number of streams

mean() [1/2]

void MLCommon::Stats::opg::mean	(	const raft::handle_t &	handle,
		Matrix::Data< double > &	out,
		const std::vector< Matrix::Data< double > * > &	in,
		const Matrix::PartDescriptor &	inDesc,
		cudaStream_t *	streams,
		int	n_streams
	)

mean() [2/2]

void MLCommon::Stats::opg::mean	(	const raft::handle_t &	handle,
		Matrix::Data< float > &	out,
		const std::vector< Matrix::Data< float > * > &	in,
		const Matrix::PartDescriptor &	inDesc,
		cudaStream_t *	streams,
		int	n_streams
	)

performs MNMG mean calculation.

Parameters

[in]	handle	cuML handle object
[out]	out	mean of every column of the input
[in]	in	all the data partitions
[in]	inDesc	MNMG description of the input data
[in]	streams	cuda streams
[in]	n_streams	number of streams

mean_add() [1/2]

void MLCommon::Stats::opg::mean_add	(	const std::vector< Matrix::Data< double > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		const Matrix::Data< double > &	mu,
		const raft::comms::comms_t &	comm,
		cudaStream_t *	streams,
		int	n_streams
	)

performs MNMG mean add calculation.

Parameters

[in,out]	data	the data that mean of every column is added
[in]	dataDesc	MNMG description of the input data
[in]	mu	mean of every column in data
[in]	comm	communicator
[in]	streams	cuda streams
[in]	n_streams	number of streams

mean_add() [2/2]

void MLCommon::Stats::opg::mean_add	(	const std::vector< Matrix::Data< float > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		const Matrix::Data< float > &	mu,
		const raft::comms::comms_t &	comm,
		cudaStream_t *	streams,
		int	n_streams
	)

mean_center() [1/2]

void MLCommon::Stats::opg::mean_center	(	const std::vector< Matrix::Data< double > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		const Matrix::Data< double > &	mu,
		const raft::comms::comms_t &	comm,
		cudaStream_t *	streams,
		int	n_streams
	)

performs MNMG mean subtraction calculation.

Parameters

[in,out]	data	the data that mean of every column is added
[in]	dataDesc	MNMG description of the input data
[in]	mu	mean of every column in data
[in]	comm	communicator
[in]	streams	cuda streams
[in]	n_streams	number of streams

mean_center() [2/2]

void MLCommon::Stats::opg::mean_center	(	const std::vector< Matrix::Data< float > * > &	data,
		const Matrix::PartDescriptor &	dataDesc,
		const Matrix::Data< float > &	mu,
		const raft::comms::comms_t &	comm,
		cudaStream_t *	streams,
		int	n_streams
	)

var() [1/2]

void MLCommon::Stats::opg::var	(	const raft::handle_t &	handle,
		Matrix::Data< double > &	out,
		const std::vector< Matrix::Data< double > * > &	in,
		const Matrix::PartDescriptor &	inDesc,
		const double *	mu,
		cudaStream_t *	streams,
		int	n_streams
	)

var() [2/2]

void MLCommon::Stats::opg::var	(	const raft::handle_t &	handle,
		Matrix::Data< float > &	out,
		const std::vector< Matrix::Data< float > * > &	in,
		const Matrix::PartDescriptor &	inDesc,
		const float *	mu,
		cudaStream_t *	streams,
		int	n_streams
	)

performs MNMG var calculation.

Parameters

[in]	handle	cuML handle object
[out]	out	var of every column of the input
[in]	in	all the data partitions
[in]	inDesc	MNMG description of the input data
[in]	mu	mean of every column in data
[in]	streams	cuda streams
[in]	n_streams	number of streams