Open3D (C++ API)  0.16.0
UnaryEW.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// The MIT License (MIT)
5//
6// Copyright (c) 2018-2021 www.open3d.org
7//
8// Permission is hereby granted, free of charge, to any person obtaining a copy
9// of this software and associated documentation files (the "Software"), to deal
10// in the Software without restriction, including without limitation the rights
11// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12// copies of the Software, and to permit persons to whom the Software is
13// furnished to do so, subject to the following conditions:
14//
15// The above copyright notice and this permission notice shall be included in
16// all copies or substantial portions of the Software.
17//
18// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
21// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
23// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
24// IN THE SOFTWARE.
25// ----------------------------------------------------------------------------
26
27#pragma once
28
29#include "open3d/core/Tensor.h"
31
32namespace open3d {
33namespace core {
34namespace kernel {
35
36enum class UnaryEWOpCode {
37 Sqrt,
38 Sin,
39 Cos,
40 Neg,
41 Exp,
42 Abs,
43 IsNan,
44 IsInf,
46 Floor,
47 Ceil,
48 Round,
49 Trunc,
51};
52
53void UnaryEW(const Tensor& src, Tensor& dst, UnaryEWOpCode op_code);
54
55void UnaryEWCPU(const Tensor& src, Tensor& dst, UnaryEWOpCode op_code);
56
57#ifdef BUILD_CUDA_MODULE
58void UnaryEWCUDA(const Tensor& src, Tensor& dst, UnaryEWOpCode op_code);
59#endif
60
61// Copy is separated from other unary ops since it support cross-device copy and
62// dtype casting.
63void Copy(const Tensor& src, Tensor& dst);
64
65void CopyCPU(const Tensor& src, Tensor& dst);
66
67#ifdef BUILD_CUDA_MODULE
68void CopyCUDA(const Tensor& src, Tensor& dst);
69#endif
70
71} // namespace kernel
72} // namespace core
73} // namespace open3d
Definition: Tensor.h:51
UnaryEWOpCode
Definition: UnaryEW.h:36
void Copy(const Tensor &src, Tensor &dst)
Definition: UnaryEW.cpp:65
void CopyCPU(const Tensor &src, Tensor &dst)
Definition: UnaryEWCPU.cpp:182
void UnaryEWCPU(const Tensor &src, Tensor &dst, UnaryEWOpCode op_code)
Definition: UnaryEWCPU.cpp:225
void UnaryEW(const Tensor &src, Tensor &dst, UnaryEWOpCode op_code)
Definition: UnaryEW.cpp:37
Definition: PinholeCameraIntrinsic.cpp:35