Searched refs:LinearAlgebraOp (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | linalg_ops_common.cc | 33 void LinearAlgebraOp<InputScalar, OutputScalar>::ValidateSingleMatrix( in ValidateSingleMatrix()
44 void LinearAlgebraOp<InputScalar, OutputScalar>::ValidateSingleSquareMatrix( in ValidateSingleSquareMatrix()
55 void LinearAlgebraOp<InputScalar, OutputScalar>::ValidateSolver( in ValidateSolver()
72 void LinearAlgebraOp<InputScalar, OutputScalar>::ValidateSquareSolver( in ValidateSquareSolver()
89 void LinearAlgebraOp<InputScalar, OutputScalar>::Compute( in Compute()
117 void LinearAlgebraOp<InputScalar, OutputScalar>::AnalyzeInputs( in AnalyzeInputs()
161 void LinearAlgebraOp<InputScalar, OutputScalar>::PrepareOutputs( in PrepareOutputs()
220 void LinearAlgebraOp<InputScalar, OutputScalar>::ComputeTensorSlice( in ComputeTensorSlice()
252 template class LinearAlgebraOp<float>; variable
253 template class LinearAlgebraOp<double>; variable
[all …]
|
| D | linalg_ops_common.h | 39 class LinearAlgebraOp : public OpKernel {
41 explicit LinearAlgebraOp(OpKernelConstruction* context) : OpKernel(context) {} in LinearAlgebraOp() function
191 extern template class LinearAlgebraOp<float>;
192 extern template class LinearAlgebraOp<double>;
193 extern template class LinearAlgebraOp<complex64>;
194 extern template class LinearAlgebraOp<complex128>;
199 typedef LinearAlgebraOp<Scalar> Base; \
|
| D | self_adjoint_eig_op.cc | 32 class SelfAdjointEigOp : public LinearAlgebraOp<Scalar> {
34 typedef LinearAlgebraOp<Scalar> Base;
|
| D | self_adjoint_eig_v2_op_impl.h | 35 class SelfAdjointEigV2Op : public LinearAlgebraOp<Scalar> {
37 typedef LinearAlgebraOp<Scalar> Base;
|
| D | eig_op_impl.h | 35 class EigOp : public LinearAlgebraOp<InputScalar, OutputScalar> {
37 typedef LinearAlgebraOp<InputScalar, OutputScalar> Base;
|
| D | svd_op_impl.h | 39 class SvdOp : public LinearAlgebraOp<Scalar> {
41 typedef LinearAlgebraOp<Scalar> Base;
|
| D | matrix_solve_ls_op_impl.h | 35 class MatrixSolveLsOp : public LinearAlgebraOp<Scalar> {
37 typedef LinearAlgebraOp<Scalar> Base;
|
| D | cholesky_grad.cc | 26 class CholeskyGrad : public LinearAlgebraOp<Scalar> {
28 typedef LinearAlgebraOp<Scalar> Base;
|
| D | matrix_logarithm_op.cc | 32 class MatrixLogarithmOp : public LinearAlgebraOp<Scalar> {
|
| D | matrix_square_root_op.cc | 32 class MatrixSquareRootOp : public LinearAlgebraOp<Scalar> {
|
| D | matrix_exponential_op.cc | 32 class MatrixExponentialOp : public LinearAlgebraOp<Scalar> {
|
| D | qr_op_impl.h | 54 class QrOp : public LinearAlgebraOp<Scalar> {
56 typedef LinearAlgebraOp<Scalar> Base;
|
| D | tridiagonal_matmul_op.cc | 30 class TridiagonalMatMulOp : public LinearAlgebraOp<Scalar> {
|
| D | determinant_op.cc | 78 class LogDeterminantOp : public LinearAlgebraOp<Scalar> {
102 class DeterminantOp : public LinearAlgebraOp<Scalar> {
|
| D | cholesky_op.cc | 46 class CholeskyOp : public LinearAlgebraOp<Scalar> {
|
| D | matrix_inverse_op.cc | 43 class MatrixInverseOp : public LinearAlgebraOp<Scalar> {
|
| D | tridiagonal_solve_op.cc | 38 class TridiagonalSolveOp : public LinearAlgebraOp<Scalar> {
|
| D | matrix_solve_op.cc | 45 class MatrixSolveOp : public LinearAlgebraOp<Scalar> {
|
| D | tridiagonal_solve_op_gpu.cu.cc | 96 class TridiagonalSolveOpGpuLinalg : public LinearAlgebraOp<Scalar> {
|