Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dbanded_triangular_solve_op.cc42 struct SequentialBandedTriangularSolveKernel { struct
43 using Matrix =
45 using ConstMatrixMap = Eigen::Map<const Matrix>;
46 using MatrixMap = Eigen::Map<Matrix>;
47 using RealScalar = typename Eigen::NumTraits<Scalar>::Real;
49 static ConstMatrixMap ConstTensorSliceToEigenMatrix(const Tensor& t, in ConstTensorSliceToEigenMatrix()
56 static MatrixMap TensorSliceToEigenMatrix(Tensor* t, int slice) { in TensorSliceToEigenMatrix()
62 static void Run(const Tensor& in_x, const Tensor& in_y, bool lower, in Run()