|
Open3D (C++ API)
0.20.0
|
#include <c10/xpu/XPUStream.h>#include "open3d/ml/contrib/TrilinearDevoxelizeSYCL.h"#include "open3d/ml/pytorch/pvcnn/TrilinearDevoxelizeKernel.h"Functions | |
| void | TrilinearDevoxelizeSYCLLauncher (int b, int c, int n, int r, int r2, int r3, bool is_training, const float *coords, const float *feat, int *inds, float *wgts, float *outs) |
| void | TrilinearDevoxelizeGradSYCLLauncher (int b, int c, int n, int r3, const int *inds, const float *wgts, const float *grad_y, float *grad_x) |
| void TrilinearDevoxelizeGradSYCLLauncher | ( | int | b, |
| int | c, | ||
| int | n, | ||
| int | r3, | ||
| const int * | inds, | ||
| const float * | wgts, | ||
| const float * | grad_y, | ||
| float * | grad_x | ||
| ) |
| void TrilinearDevoxelizeSYCLLauncher | ( | int | b, |
| int | c, | ||
| int | n, | ||
| int | r, | ||
| int | r2, | ||
| int | r3, | ||
| bool | is_training, | ||
| const float * | coords, | ||
| const float * | feat, | ||
| int * | inds, | ||
| float * | wgts, | ||
| float * | outs | ||
| ) |