mean_center.hpp File Reference

#include "../matrix/data.hpp"
#include "../matrix/part_descriptor.hpp"
#include <raft/core/comms.hpp>

Include dependency graph for mean_center.hpp:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::Stats
	MLCommon::Stats::opg

Functions
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. More...
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)
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. More...
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)