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

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

#include <torch/script.h>
#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/InvertNeighborsListOps.h"
#include "open3d/ml/pytorch/misc/ReduceSubarraysSumOps.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvBackpropFilterOpKernel.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvOpKernel.h"
#include "open3d/ml/pytorch/sparse_conv/SparseConvTransposeOpKernel.h"

Data Structures

class  SparseConvFunction
 

Macros

#define FN_PARAMETERS
 
#define CALL(feat_t, out_t, index_t, kernel_index_t, fn)
 
#define CALL(feat_t, out_t, index_t, kernel_index_t, fn_suffix)
 

Functions

torch::Tensor SparseConv (const torch::Tensor &filters, const torch::Tensor &inp_features, const torch::Tensor &inp_importance, const torch::Tensor &neighbors_index, const torch::Tensor &neighbors_kernel_index, const torch::Tensor &neighbors_importance, const torch::Tensor &neighbors_row_splits, const bool normalize, const int64_t max_temp_mem_MB)
 

Macro Definition Documentation

◆ CALL [1/2]

#define CALL (   feat_t,
  out_t,
  index_t,
  kernel_index_t,
  fn 
)
Value:
if (CompareTorchDtype<feat_t>(feat_dtype) && \
CompareTorchDtype<index_t>(index_dtype) && \
CompareTorchDtype<kernel_index_t>(kernel_index_dtype)) { \
fn<feat_t, out_t, index_t, kernel_index_t>(FN_PARAMETERS); \
return out_features; \
}
#define FN_PARAMETERS

◆ CALL [2/2]

#define CALL (   feat_t,
  out_t,
  index_t,
  kernel_index_t,
  fn_suffix 
)

◆ FN_PARAMETERS

#define FN_PARAMETERS
Value:
filters, inp_features, inp_importance, neighbors_index, \
neighbors_kernel_index, neighbors_importance, \
neighbors_row_splits, normalize, max_temp_mem_MB, out_features

Function Documentation

◆ SparseConv()

torch::Tensor SparseConv ( const torch::Tensor &  filters,
const torch::Tensor &  inp_features,
const torch::Tensor &  inp_importance,
const torch::Tensor &  neighbors_index,
const torch::Tensor &  neighbors_kernel_index,
const torch::Tensor &  neighbors_importance,
const torch::Tensor &  neighbors_row_splits,
const bool  normalize,
const int64_t  max_temp_mem_MB 
)