Open3D (C++ API)  0.12.0
Macros | Functions
BuildSpatialHashTableOps.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"

Macros

#define FN_PARAMETERS
 
#define CALL(type, fn)
 

Functions

template<class T >
void BuildSpatialHashTableCPU (const torch::Tensor &points, double radius, const torch::Tensor &points_row_splits, const std::vector< uint32_t > &hash_table_splits, torch::Tensor &hash_table_index, torch::Tensor &hash_table_cell_splits)
 
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor > BuildSpatialHashTable (torch::Tensor points, double radius, torch::Tensor points_row_splits, double hash_table_size_factor, int64_t max_hash_table_size)
 

Macro Definition Documentation

◆ CALL

#define CALL (   type,
  fn 
)
Value:
if (CompareTorchDtype<type>(point_type)) { \
fn<type>(FN_PARAMETERS); \
return std::make_tuple(hash_table_index, hash_table_cell_splits, \
out_hash_table_splits); \
}
#define FN_PARAMETERS

◆ FN_PARAMETERS

#define FN_PARAMETERS
Value:
points, radius, points_row_splits, hash_table_splits, hash_table_index, \
hash_table_cell_splits
int points
Definition: FilePCD.cpp:73

Function Documentation

◆ BuildSpatialHashTable()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> BuildSpatialHashTable ( torch::Tensor  points,
double  radius,
torch::Tensor  points_row_splits,
double  hash_table_size_factor,
int64_t  max_hash_table_size 
)

◆ BuildSpatialHashTableCPU()

template<class T >
void BuildSpatialHashTableCPU ( const torch::Tensor &  points,
double  radius,
const torch::Tensor &  points_row_splits,
const std::vector< uint32_t > &  hash_table_splits,
torch::Tensor &  hash_table_index,
torch::Tensor &  hash_table_cell_splits 
)