Open3D (C++ API)  0.19.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-2024 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_SYCL_MODULE
24void 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
42void 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
59void 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
open3d::core::Device
Definition Device.h:18
open3d::core::Dtype
Definition Dtype.h:20
open3d::core::Tensor
Definition Tensor.h:32
open3d::core
Definition AdvancedIndexing.cpp:15
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.15.0