Loading [MathJax]/extensions/TeX/AMSsymbols.js
Open3D (C++ API)  0.19.0
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
AddMM.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"
11 
12 namespace open3d {
13 namespace core {
14 
20 void AddMM(
21  const Tensor& A, const Tensor& B, Tensor& C, double alpha, double beta);
22 
23 #ifdef BUILD_SYCL_MODULE
24 void AddMMSYCL(void* A_data,
25  void* B_data,
26  void* C_data,
27  int64_t m,
28  int64_t k,
29  int64_t n,
30  double alpha,
31  double beta,
32  bool gemmTrA,
33  bool gemmTrB,
34  int lda,
35  int ldb,
36  int ldc,
37  Dtype dtype,
38  const Device& device);
39 #endif
40 
41 #ifdef BUILD_CUDA_MODULE
42 void AddMMCUDA(void* A_data,
43  void* B_data,
44  void* C_data,
45  int64_t m,
46  int64_t k,
47  int64_t n,
48  double alpha,
49  double beta,
50  bool gemmTrA,
51  bool gemmTrB,
52  int lda,
53  int ldb,
54  int ldc,
55  Dtype dtype,
56  const Device& device);
57 #endif
58 
59 void AddMMCPU(void* A_data,
60  void* B_data,
61  void* C_data,
62  int64_t m,
63  int64_t k,
64  int64_t n,
65  double alpha,
66  double beta,
67  bool gemmTrA,
68  bool gemmTrB,
69  int lda,
70  int ldb,
71  int ldc,
72  Dtype dtype);
73 
74 } // namespace core
75 } // namespace open3d
B
Eigen::Matrix3d B
Definition: PointCloudPlanarPatchDetection.cpp:519
open3d::core::AddMMCUDA
void AddMMCUDA(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, double alpha, double beta, bool gemmTrA, bool gemmTrB, int lda, int ldb, int ldc, Dtype dtype, const Device &device)
Definition: AddMMCUDA.cpp:16
open3d::core::AddMMCPU
void AddMMCPU(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, double alpha, double beta, bool gemmTrA, bool gemmTrB, int lda, int ldb, int ldc, Dtype dtype)
Definition: AddMMCPU.cpp:16
open3d::core::AddMM
void AddMM(const Tensor &A, const Tensor &B, Tensor &output, double alpha, double beta)
Definition: AddMM.cpp:17
open3d::core::AddMMSYCL
void AddMMSYCL(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, double alpha, double beta, bool gemmTrA, bool gemmTrB, int lda, int ldb, int ldc, Dtype dtype, const Device &device)
Definition: AddMMSYCL.cpp:19
open3d
Definition: PinholeCameraIntrinsic.cpp:16
Generated by doxygen 1.9.1