Home
last modified time | relevance | path

Searched refs:ComputeMatrix (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_logarithm_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixLogarithmOp
Dmatrix_square_root_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixSquareRootOp
Dmatrix_exponential_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixExponentialOp
Dself_adjoint_eig_op.cc50 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::SelfAdjointEigOp
Dself_adjoint_eig_v2_op_impl.h59 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
Deig_op_impl.h64 void ComputeMatrix(OpKernelContext* context, in ComputeMatrix() function
Dsvd_op_impl.h84 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
Dtridiagonal_matmul_op.cc83 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalMatMulOp
Dmatrix_solve_ls_op_impl.h76 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
Dcholesky_grad.cc60 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::CholeskyGrad
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
Dcholesky_op.cc52 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::CholeskyOp
Dlinalg_ops_common.h149 virtual void ComputeMatrix(OpKernelContext* context,
Dmatrix_inverse_op.cc51 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixInverseOp
Dtridiagonal_solve_op.cc100 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalSolveOp
Dlinalg_ops_common.cc248 ComputeMatrix(context, matrix_inputs, &matrix_outputs); in ComputeTensorSlice()
Dqr_op_impl.h101 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function
Dmatrix_solve_op.cc75 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::MatrixSolveOp
Dtridiagonal_solve_op_gpu.cu.cc135 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix() function in tensorflow::TridiagonalSolveOpGpuLinalg