Open3D (C++ API)  0.17.0
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
12namespace open3d {
13namespace core {
14
16void Matmul(const Tensor& A, const Tensor& B, Tensor& C);
17
18#ifdef BUILD_CUDA_MODULE
19void 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
28void 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:506
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