Open3D (C++ API)  0.19.0
Loading...
Searching...
No Matches
IndexGetSet.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// Copyright (c) 2018-2024 www.open3d.org
5// SPDX-License-Identifier: MIT
6// ----------------------------------------------------------------------------
7
8#pragma once
9
10#include "open3d/core/Tensor.h"
12
13namespace open3d {
14namespace core {
15namespace kernel {
16
17void IndexGet(const Tensor& src,
18 Tensor& dst,
19 const std::vector<Tensor>& index_tensors,
20 const SizeVector& indexed_shape,
21 const SizeVector& indexed_strides);
22
23void IndexGetCPU(const Tensor& src,
24 Tensor& dst,
25 const std::vector<Tensor>& index_tensors,
26 const SizeVector& indexed_shape,
27 const SizeVector& indexed_strides);
28
29#ifdef BUILD_SYCL_MODULE
30void IndexGetSYCL(const Tensor& src,
31 Tensor& dst,
32 const std::vector<Tensor>& index_tensors,
33 const SizeVector& indexed_shape,
34 const SizeVector& indexed_strides);
35#endif
36
37#ifdef BUILD_CUDA_MODULE
38void IndexGetCUDA(const Tensor& src,
39 Tensor& dst,
40 const std::vector<Tensor>& index_tensors,
41 const SizeVector& indexed_shape,
42 const SizeVector& indexed_strides);
43#endif
44
45void IndexSet(const Tensor& src,
46 Tensor& dst,
47 const std::vector<Tensor>& index_tensors,
48 const SizeVector& indexed_shape,
49 const SizeVector& indexed_strides);
50
51void IndexSetCPU(const Tensor& src,
52 Tensor& dst,
53 const std::vector<Tensor>& index_tensors,
54 const SizeVector& indexed_shape,
55 const SizeVector& indexed_strides);
56
57#ifdef BUILD_SYCL_MODULE
58void IndexSetSYCL(const Tensor& src,
59 Tensor& dst,
60 const std::vector<Tensor>& index_tensors,
61 const SizeVector& indexed_shape,
62 const SizeVector& indexed_strides);
63#endif
64
65#ifdef BUILD_CUDA_MODULE
66void IndexSetCUDA(const Tensor& src,
67 Tensor& dst,
68 const std::vector<Tensor>& index_tensors,
69 const SizeVector& indexed_shape,
70 const SizeVector& indexed_strides);
71#endif
72
73} // namespace kernel
74} // namespace core
75} // namespace open3d
void IndexGet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSet.cpp:21
void IndexGetSYCL(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSetSYCL.cpp:18
void IndexSetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSetCPU.cpp:61
void IndexSet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSet.cpp:51
void IndexSetSYCL(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSetSYCL.cpp:46
void IndexGetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition IndexGetSetCPU.cpp:41
Definition PinholeCameraIntrinsic.cpp:16