Searched refs:MatrixInverseOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_inverse_op.cc | 43 class MatrixInverseOp : public LinearAlgebraOp<Scalar> { class
47 explicit MatrixInverseOp(OpKernelConstruction* context) : Base(context) { in MatrixInverseOp() function in tensorflow::MatrixInverseOp
83 TF_DISALLOW_COPY_AND_ASSIGN(MatrixInverseOp);
278 REGISTER_LINALG_OP("MatrixInverse", (MatrixInverseOp<float>), float);
279 REGISTER_LINALG_OP("MatrixInverse", (MatrixInverseOp<double>), double);
280 REGISTER_LINALG_OP("MatrixInverse", (MatrixInverseOp<complex64>), complex64);
281 REGISTER_LINALG_OP("MatrixInverse", (MatrixInverseOp<complex128>), complex128);
282 REGISTER_LINALG_OP("BatchMatrixInverse", (MatrixInverseOp<float>), float);
283 REGISTER_LINALG_OP("BatchMatrixInverse", (MatrixInverseOp<double>), double);
|