Open3D (C++ API)  0.18.0+80ae047
Matmul.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 
16 void Matmul(const Tensor& A, const Tensor& B, Tensor& C);
17 
18 #ifdef BUILD_CUDA_MODULE
19 void MatmulCUDA(void* A_data,
20  void* B_data,
21  void* C_data,
22  int64_t m,
23  int64_t k,
24  int64_t n,
25  Dtype dtype,
26  const Device& device);
27 #endif
28 void MatmulCPU(void* A_data,
29  void* B_data,
30  void* C_data,
31  int64_t m,
32  int64_t k,
33  int64_t n,
34  Dtype dtype);
35 } // namespace core
36 } // namespace open3d
Eigen::Matrix3d B
Definition: PointCloudPlanarPatchDetection.cpp:512
void Matmul(const Tensor &A, const Tensor &B, Tensor &output)
Computes matrix multiplication C = AB.
Definition: Matmul.cpp:17
void MatmulCPU(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype)
Definition: MatmulCPU.cpp:14
void MatmulCUDA(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype, const Device &device)
Definition: MatmulCUDA.cpp:16
Definition: PinholeCameraIntrinsic.cpp:16