Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Ddeterminant_op.cc89 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::LogDeterminantOp
113 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::DeterminantOp
H A Dcholesky_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::CholeskyOp
H A Dmatrix_exponential_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixExponentialOp
H A Dmatrix_logarithm_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixLogarithmOp
H A Dmatrix_square_root_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixSquareRootOp
H A Dself_adjoint_eig_v2_op_impl.h59 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
H A Dself_adjoint_eig_op.cc50 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::SelfAdjointEigOp
H A Deig_op_impl.h64 void ComputeMatrix(OpKernelContext* context, in ComputeMatrix() function
H A Dsvd_op_impl.h84 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
H A Dmatrix_solve_ls_op_impl.h76 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
H A Dtridiagonal_matmul_op.cc83 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalMatMulOp
H A Dmatrix_inverse_op.cc51 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixInverseOp
H A Dcholesky_grad.cc60 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::CholeskyGrad
H A Dmatrix_solve_op.cc75 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixSolveOp
H A Dqr_op_impl.h101 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
H A Dtridiagonal_solve_op.cc110 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalSolveOp
H A Dtridiagonal_solve_op_gpu.cu.cc141 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalSolveOpGpuLinalg