|
Open3D (C++ API)
0.20.0
|
#include <c10/xpu/XPUStream.h>#include "open3d/ml/contrib/InterpolatePointsSYCL.h"#include "open3d/ml/pytorch/pointnet/InterpolateKernel.h"Functions | |
| void | three_nn_launcher_sycl (int b, int n, int m, const float *unknown, const float *known, float *dist2, int *idx) |
| void | three_interpolate_launcher_sycl (int b, int c, int m, int n, const float *points, const int *idx, const float *weight, float *out) |
| void | three_interpolate_grad_launcher_sycl (int b, int c, int n, int m, const float *grad_out, const int *idx, const float *weight, float *grad_points) |
| void three_interpolate_grad_launcher_sycl | ( | int | b, |
| int | c, | ||
| int | n, | ||
| int | m, | ||
| const float * | grad_out, | ||
| const int * | idx, | ||
| const float * | weight, | ||
| float * | grad_points | ||
| ) |
| void three_interpolate_launcher_sycl | ( | int | b, |
| int | c, | ||
| int | m, | ||
| int | n, | ||
| const float * | points, | ||
| const int * | idx, | ||
| const float * | weight, | ||
| float * | out | ||
| ) |
| void three_nn_launcher_sycl | ( | int | b, |
| int | n, | ||
| int | m, | ||
| const float * | unknown, | ||
| const float * | known, | ||
| float * | dist2, | ||
| int * | idx | ||
| ) |