tsvd.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2018-2025, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 
6 #pragma once
7 
8 #include "params.hpp"
9 
10 namespace raft {
11 class handle_t;
12 }
13 
14 namespace ML {
15 
16 void tsvdFit(raft::handle_t& handle,
17  float* input,
18  float* components,
19  float* singular_vals,
20  const paramsTSVD& prms,
21  bool flip_signs_based_on_U);
22 void tsvdFit(raft::handle_t& handle,
23  double* input,
24  double* components,
25  double* singular_vals,
26  const paramsTSVD& prms,
27  bool flip_signs_based_on_U);
28 void tsvdInverseTransform(raft::handle_t& handle,
29  float* trans_input,
30  float* components,
31  float* input,
32  const paramsTSVD& prms);
33 void tsvdInverseTransform(raft::handle_t& handle,
34  double* trans_input,
35  double* components,
36  double* input,
37  const paramsTSVD& prms);
38 void tsvdTransform(raft::handle_t& handle,
39  float* input,
40  float* components,
41  float* trans_input,
42  const paramsTSVD& prms);
43 void tsvdTransform(raft::handle_t& handle,
44  double* input,
45  double* components,
46  double* trans_input,
47  const paramsTSVD& prms);
48 void tsvdFitTransform(raft::handle_t& handle,
49  float* input,
50  float* trans_input,
51  float* components,
52  float* explained_var,
53  float* explained_var_ratio,
54  float* singular_vals,
55  const paramsTSVD& prms,
56  bool flip_signs_based_on_U);
57 void tsvdFitTransform(raft::handle_t& handle,
58  double* input,
59  double* trans_input,
60  double* components,
61  double* explained_var,
62  double* explained_var_ratio,
63  double* singular_vals,
64  const paramsTSVD& prms,
65  bool flip_signs_based_on_U);
66 
67 } // namespace ML
ML::paramsTSVDTemplate
Definition: params.hpp:39
ML
Definition: dbscan.hpp:18
ML::tsvdTransform
void tsvdTransform(raft::handle_t &handle, float *input, float *components, float *trans_input, const paramsTSVD &prms)
ML::tsvdFitTransform
void tsvdFitTransform(raft::handle_t &handle, float *input, float *trans_input, float *components, float *explained_var, float *explained_var_ratio, float *singular_vals, const paramsTSVD &prms, bool flip_signs_based_on_U)
ML::tsvdFit
void tsvdFit(raft::handle_t &handle, float *input, float *components, float *singular_vals, const paramsTSVD &prms, bool flip_signs_based_on_U)
ML::tsvdInverseTransform
void tsvdInverseTransform(raft::handle_t &handle, float *trans_input, float *components, float *input, const paramsTSVD &prms)
raft
Definition: dbscan.hpp:14
Generated by doxygen 1.9.1