Distance

1
enum class DistanceType : int {
2
  L2Expanded = 0,
3
  CosineExpanded = 2,
4
  L1 = 3,
5
  L2Unexpanded = 4,
6
  InnerProduct = 6,
7
  Linf = 7,
8
  Canberra = 8,
9
  LpUnexpanded = 9,
10
  CorrelationExpanded = 10,
11
  JaccardExpanded = 11,
12
  HellingerExpanded = 12,
13
  Haversine = 13,
14
  BrayCurtis = 14,
15
  JensenShannon = 15,
16
  HammingUnexpanded = 16,
17
  KLDivergence = 17,
18
  RusselRaoExpanded = 18,
19
  DiceExpanded = 19,
20
  BitwiseHamming = 20,
21
  Precomputed = 100,
22
  CustomUDF = 101
23
};

1
enum class DensityKernelType : int {
2
  Gaussian = 0,
3
  Tophat = 1,
4
  Epanechnikov = 2,
5
  Exponential = 3,
6
  Linear = 4,
7
  Cosine = 5
8
};

1
struct KernelParams {
2
  KernelType kernel;
3
  int degree;
4
  double gamma;
5
  double coef0;
6
};

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const float, std::int64_t, raft::layout_c_contiguous> const x,
4
raft::device_matrix_view<const float, std::int64_t, raft::layout_c_contiguous> const y,
5
raft::device_matrix_view<float, std::int64_t, raft::layout_c_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
float metric_arg = 2.0f);

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const double, std::int64_t, raft::layout_c_contiguous> const x,
4
raft::device_matrix_view<const double, std::int64_t, raft::layout_c_contiguous> const y,
5
raft::device_matrix_view<double, std::int64_t, raft::layout_c_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
double metric_arg = 2.0f);

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const half, std::int64_t, raft::layout_c_contiguous> const x,
4
raft::device_matrix_view<const half, std::int64_t, raft::layout_c_contiguous> const y,
5
raft::device_matrix_view<float, std::int64_t, raft::layout_c_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
float metric_arg = 2.0f);

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const float, std::int64_t, raft::layout_f_contiguous> const x,
4
raft::device_matrix_view<const float, std::int64_t, raft::layout_f_contiguous> const y,
5
raft::device_matrix_view<float, std::int64_t, raft::layout_f_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
float metric_arg = 2.0f);

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const double, std::int64_t, raft::layout_f_contiguous> const x,
4
raft::device_matrix_view<const double, std::int64_t, raft::layout_f_contiguous> const y,
5
raft::device_matrix_view<double, std::int64_t, raft::layout_f_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
double metric_arg = 2.0f);

1
void pairwise_distance(
2
raft::resources const& handle,
3
raft::device_matrix_view<const half, std::int64_t, raft::layout_f_contiguous> const x,
4
raft::device_matrix_view<const half, std::int64_t, raft::layout_f_contiguous> const y,
5
raft::device_matrix_view<float, std::int64_t, raft::layout_f_contiguous> dist,
6
cuvs::distance::DistanceType metric,
7
float metric_arg = 2.0f);

1
void pairwise_distance(raft::resources const& handle,
2
raft::device_csr_matrix_view<const float, int, int, int> x,
3
raft::device_csr_matrix_view<const float, int, int, int> y,
4
raft::device_matrix_view<float, int, raft::row_major> dist,
5
cuvs::distance::DistanceType metric,
6
float metric_arg = 2.0f);

1
void pairwise_distance(raft::resources const& handle,
2
raft::device_csr_matrix_view<const double, int, int, int> x,
3
raft::device_csr_matrix_view<const double, int, int, int> y,
4
raft::device_matrix_view<double, int, raft::row_major> dist,
5
cuvs::distance::DistanceType metric,
6
float metric_arg = 2.0f);