norm.hpp File Reference

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

Include dependency graph for norm.hpp:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::LinAlg
	MLCommon::LinAlg::opg

Functions
void	MLCommon::LinAlg::opg::colNorm2 (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)
	performs MNMG Least squares calculation. More...
void	MLCommon::LinAlg::opg::colNorm2 (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)
void	MLCommon::LinAlg::opg::colNorm2NoSeq (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	MLCommon::LinAlg::opg::colNorm2NoSeq (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)