35 void Matmul(
const Tensor& A,
const Tensor& B, Tensor& C);
37 #ifdef BUILD_CUDA_MODULE 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:33
void Matmul(const Tensor &A, const Tensor &B, Tensor &output)
Computes matrix multiplication C = AB.
Definition: Matmul.cpp:34
Definition: PinholeCameraIntrinsic.cpp:35
void MatmulCUDA(void *A_data, void *B_data, void *C_data, int64_t m, int64_t k, int64_t n, Dtype dtype)
Definition: MatmulCUDA.cpp:35