Open3D (C++ API)  0.18.0+252c867
AddMM.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 "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_CUDA_MODULE
24 void AddMMCUDA(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 void AddMMCPU(void* A_data,
42  void* B_data,
43  void* C_data,
44  int64_t m,
45  int64_t k,
46  int64_t n,
47  double alpha,
48  double beta,
49  bool gemmTrA,
50  bool gemmTrB,
51  int lda,
52  int ldb,
53  int ldc,
54  Dtype dtype);
55 
56 } // namespace core
57 } // namespace open3d
Eigen::Matrix3d B
Definition: PointCloudPlanarPatchDetection.cpp:512
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
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
void AddMM(const Tensor &A, const Tensor &B, Tensor &output, double alpha, double beta)
Definition: AddMM.cpp:17
Definition: PinholeCameraIntrinsic.cpp:16