Open3D (C++ API)  0.18.0+252c867
Macros | Functions
ContinuousConvOpKernel.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#include <vector>
#include "open3d/ml/impl/continuous_conv/ContinuousConv.h"
#include "torch/script.h"

Macros

#define INSTANTIATE(TFeat, TOut, TReal, TIndex)
 

Functions

template<class TFeat , class TOut , 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)
 

Macro Definition Documentation

◆ INSTANTIATE

#define INSTANTIATE (   TFeat,
  TOut,
  TReal,
  TIndex 
)
Value:
template void ContinuousConvCPU<TFeat, TOut, TReal, TIndex>( \
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);
int offset
Definition: FilePCD.cpp:45
InterpolationMode
Definition: ContinuousConvTypes.h:18
CoordinateMapping
Definition: ContinuousConvTypes.h:26

Function Documentation

◆ ContinuousConvCPU()

template<class TFeat , class TOut , 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 
)