Open3D (C++ API)  0.18.0+252c867
ContinuousConvOpKernel.h
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // - Open3D: www.open3d.org -
3 // ----------------------------------------------------------------------------
4 // Copyright (c) 2018-2023 www.open3d.org
5 // SPDX-License-Identifier: MIT
6 // ----------------------------------------------------------------------------
7 //
8 #pragma once
9 
10 #include <vector>
11 
13 #include "torch/script.h"
14 
15 template <class TFeat, class TOut, class TReal, class TIndex>
17  const torch::Tensor& filters,
18  const torch::Tensor& out_positions,
19  const torch::Tensor& extents,
20  const torch::Tensor& offset,
21  const torch::Tensor& inp_positions,
22  const torch::Tensor& inp_features,
23  const torch::Tensor& inp_importance,
24  const torch::Tensor& neighbors_index,
25  const torch::Tensor& neighbors_importance,
26  const torch::Tensor& neighbors_row_splits,
27  const bool align_corners,
28  const open3d::ml::impl::CoordinateMapping coordinate_mapping,
29  const bool normalize,
30  const open3d::ml::impl::InterpolationMode interpolation,
31  const int64_t max_temp_mem_MB,
32  torch::Tensor& out_features);
33 
34 #ifdef BUILD_CUDA_MODULE
35 template <class TFeat, class TOut, class TReal, class TIndex>
36 void ContinuousConvCUDA(
37  const torch::Tensor& filters,
38  const torch::Tensor& out_positions,
39  const torch::Tensor& extents,
40  const torch::Tensor& offset,
41  const torch::Tensor& inp_positions,
42  const torch::Tensor& inp_features,
43  const torch::Tensor& inp_importance,
44  const torch::Tensor& neighbors_index,
45  const torch::Tensor& neighbors_importance,
46  const torch::Tensor& neighbors_row_splits,
47  const bool align_corners,
48  const open3d::ml::impl::CoordinateMapping coordinate_mapping,
49  const bool normalize,
50  const open3d::ml::impl::InterpolationMode interpolation,
51  const int64_t max_temp_mem_MB,
52  torch::Tensor& out_features);
53 #endif
int offset
Definition: FilePCD.cpp:45
InterpolationMode
Definition: ContinuousConvTypes.h:18
CoordinateMapping
Definition: ContinuousConvTypes.h:26
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 open3d::ml::impl::CoordinateMapping coordinate_mapping, const bool normalize, const open3d::ml::impl::InterpolationMode interpolation, const int64_t max_temp_mem_MB, torch::Tensor &out_features)
Definition: ContinuousConvOpKernel.cpp:17