16void TriuCPU(
const Tensor& A, Tensor& output,
const int diagonal = 0);
18void TrilCPU(
const Tensor& A, Tensor& output,
const int diagonal = 0);
23 const int diagonal = 0);
25#ifdef BUILD_CUDA_MODULE
26void TriuCUDA(
const Tensor& A, Tensor& output,
const int diagonal = 0);
28void TrilCUDA(
const Tensor& A, Tensor& output,
const int diagonal = 0);
30void TriulCUDA(
const Tensor& A,
33 const int diagonal = 0);
void TrilCPU(const Tensor &A, Tensor &output, const int diagonal)
Definition TriCPU.cpp:34
void TriulCPU(const Tensor &A, Tensor &upper, Tensor &lower, const int diagonal)
Definition TriCPU.cpp:51
void TriuCPU(const Tensor &A, Tensor &output, const int diagonal)
Definition TriCPU.cpp:17
Definition PinholeCameraIntrinsic.cpp:16