Open3D (C++ API)
0.12.0
|
#include <vector>
#include "open3d/ml/impl/continuous_conv/ContinuousConv.h"
#include "torch/script.h"
Macros | |
#define | INSTANTIATE(TReal, TIndex) |
Functions | |
template<class TReal , class TIndex > | |
void | ContinuousConvCPU (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, torch::Tensor &out_features) |
#define INSTANTIATE | ( | TReal, | |
TIndex | |||
) |
void ContinuousConvCPU | ( | 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, | ||
torch::Tensor & | out_features | ||
) |