knn_sparse.hpp File Reference

#include <cuml/neighbors/knn.hpp>
#include <raft/distance/distance_types.hpp>
#include <cusparse_v2.h>

Include dependency graph for knn_sparse.hpp:

Go to the source code of this file.

Namespaces
	raft
	ML
	ML::Sparse

Functions
void	ML::Sparse::brute_force_knn (raft::handle_t &handle, const int *idx_indptr, const int *idx_indices, const float *idx_data, size_t idx_nnz, int n_idx_rows, int n_idx_cols, const int *query_indptr, const int *query_indices, const float *query_data, size_t query_nnz, int n_query_rows, int n_query_cols, int *output_indices, float *output_dists, int k, size_t batch_size_index=DEFAULT_BATCH_SIZE, size_t batch_size_query=DEFAULT_BATCH_SIZE, raft::distance::DistanceType metric=raft::distance::DistanceType::L2Expanded, float metricArg=0)

Variables
constexpr int	ML::Sparse::DEFAULT_BATCH_SIZE = 1 << 16