Open3D (C++ API)  0.17.0
Loading...
Searching...
No Matches
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
12namespace open3d {
13namespace core {
14
20void AddMM(
21 const Tensor& A, const Tensor& B, Tensor& C, double alpha, double beta);
22
23#ifdef BUILD_CUDA_MODULE
24void 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
41void 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:506
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