#include "../matrix/data.hpp"#include "../matrix/part_descriptor.hpp"#include <raft/core/comms.hpp>#include <raft/core/handle.hpp>
Include dependency graph for mm_aTa.hpp:
Go to the source code of this file.
Namespaces | |
| MLCommon | |
| MLCommon::LinAlg | |
| MLCommon::LinAlg::opg | |
Functions | |
| void | MLCommon::LinAlg::opg::mm_aTa (const raft::handle_t &handle, Matrix::Data< double > &out, const std::vector< Matrix::Data< double > * > &in1, const Matrix::PartDescriptor &in1Desc, cudaStream_t *streams, int n_streams) |
| performs MNMG A^T x A calculation. More... | |
| void | MLCommon::LinAlg::opg::mm_aTa (const raft::handle_t &handle, Matrix::Data< float > &out, const std::vector< Matrix::Data< float > * > &in1, const Matrix::PartDescriptor &in1Desc, cudaStream_t *streams, int n_streams) |
