Home
last modified time | relevance | path

Searched refs:ConstMatrixMaps (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dself_adjoint_eig_op.cc42 using ConstMatrixMaps = typename Base::ConstMatrixMaps; typedef in tensorflow::SelfAdjointEigOp
50 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dself_adjoint_eig_v2_op_impl.h47 using ConstMatrixMaps = typename Base::ConstMatrixMaps; variable
59 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dsvd_op_impl.h82 using ConstMatrixMaps = typename Base::ConstMatrixMaps; variable
84 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_solve_ls_op_impl.h47 using ConstMatrixMaps = typename Base::ConstMatrixMaps; variable
76 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dcholesky_grad.cc37 using ConstMatrixMaps = typename Base::ConstMatrixMaps; typedef in tensorflow::CholeskyGrad
60 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dlinalg_ops_common.h139 using ConstMatrixMaps = gtl::InlinedVector<ConstMatrixMap, 4>; variable
205 using ConstMatrixMaps = typename Base::ConstMatrixMaps; \
Dqr_op_impl.h99 using ConstMatrixMaps = typename Base::ConstMatrixMaps; variable
101 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_logarithm_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_square_root_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_exponential_op.cc38 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dtridiagonal_solve_op.cc42 decltype(std::declval<const ConstMatrixMaps>()[0].row(0));
100 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dtridiagonal_matmul_op.cc83 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Ddeterminant_op.cc89 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
113 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dcholesky_op.cc52 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_inverse_op.cc51 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dmatrix_solve_op.cc75 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()
Dtridiagonal_solve_op_gpu.cu.cc135 void ComputeMatrix(OpKernelContext* context, const ConstMatrixMaps& inputs, in ComputeMatrix()