|
Open3D (C++ API)
0.20.0
|
#include <c10/xpu/XPUStream.h>#include <torch/script.h>#include <vector>#include "open3d/ml/impl/continuous_conv/ContinuousConvSYCL.h"#include "open3d/ml/pytorch/TorchHelper.h"Macros | |
| #define | INSTANTIATE(TFeat, TOut, TReal, TIndex) |
Functions | |
| template<class TFeat , class TOut , class TReal , class TIndex > | |
| void | ContinuousConvSYCL (const torch::Tensor &filters, const torch::Tensor &out_positions, const torch::Tensor &extents, const torch::Tensor &offset, const torch::Tensor &inp_positions, const torch::Tensor &inp_features, const torch::Tensor &inp_importance, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const bool align_corners, const CoordinateMapping coordinate_mapping, const bool normalize, const InterpolationMode interpolation, const int64_t max_temp_mem_MB, const bool allow_tf32, torch::Tensor &out_features) |
| #define INSTANTIATE | ( | TFeat, | |
| TOut, | |||
| TReal, | |||
| TIndex | |||
| ) |
| void ContinuousConvSYCL | ( | const torch::Tensor & | filters, |
| const torch::Tensor & | out_positions, | ||
| const torch::Tensor & | extents, | ||
| const torch::Tensor & | offset, | ||
| const torch::Tensor & | inp_positions, | ||
| const torch::Tensor & | inp_features, | ||
| const torch::Tensor & | inp_importance, | ||
| const torch::Tensor & | neighbors_index, | ||
| const torch::Tensor & | neighbors_importance, | ||
| const torch::Tensor & | neighbors_row_splits, | ||
| const bool | align_corners, | ||
| const CoordinateMapping | coordinate_mapping, | ||
| const bool | normalize, | ||
| const InterpolationMode | interpolation, | ||
| const int64_t | max_temp_mem_MB, | ||
| const bool | allow_tf32, | ||
| torch::Tensor & | out_features | ||
| ) |