Home
last modified time | relevance | path

Searched defs:lu_factor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/linalg/
H A D__init__.py2271 lu_factor = _add_docstr(_linalg.linalg_lu_factor, r""" variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Ddeterminant_op_gpu.cu.cc53 int nthreads, int n, const Scalar* __restrict__ lu_factor, in DeterminantFromPivotedLUKernel()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dlinalg.h89 inline std::tuple<Tensor, Tensor> lu_factor( in lu_factor() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp1601 static void lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_… in lu_factor() function