26 lines
966 B
Plaintext
26 lines
966 B
Plaintext
|
// ----------------------------------------------------------------------------
|
||
|
|
// - Open3D: www.open3d.org -
|
||
|
|
// ----------------------------------------------------------------------------
|
||
|
|
// Copyright (c) 2018-2023 www.open3d.org
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
// ----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
namespace open3d {
|
||
|
|
namespace ml {
|
||
|
|
namespace contrib {
|
||
|
|
|
||
|
|
__global__ void ball_query_kernel(int b,
|
||
|
|
int n,
|
||
|
|
int m,
|
||
|
|
float radius,
|
||
|
|
int nsample,
|
||
|
|
const float *__restrict__ new_xyz,
|
||
|
|
const float *__restrict__ xyz,
|
||
|
|
int *__restrict__ idx);
|
||
|
|
|
||
|
|
} // namespace contrib
|
||
|
|
} // namespace ml
|
||
|
|
} // namespace open3d
|