35 void Solve(
const Tensor& A,
const Tensor& B, Tensor& X);
43 const Device& device);
45 #ifdef BUILD_CUDA_MODULE 52 const Device& device);
void SolveCPU(void *A_data, void *B_data, void *ipiv_data, int64_t n, int64_t k, Dtype dtype, const Device &device)
Definition: SolveCPU.cpp:34
void Solve(const Tensor &A, const Tensor &B, Tensor &X)
Solve AX = B with LU decomposition. A is a square matrix.
Definition: Solve.cpp:40
Definition: PinholeCameraIntrinsic.cpp:35
void SolveCUDA(void *A_data, void *B_data, void *ipiv_data, int64_t n, int64_t k, Dtype dtype, const Device &device)
Definition: SolveCUDA.cpp:38