Home
last modified time | relevance | path

Searched refs:QrOp (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dqr_op_double.cc20 REGISTER_LINALG_OP("Qr", (QrOp<double>), double);
33 QrOp<double>);
Dqr_op_float.cc20 REGISTER_LINALG_OP("Qr", (QrOp<float>), float);
33 QrOp<float>);
Dqr_op_complex128.cc20 REGISTER_LINALG_OP("Qr", (QrOp<complex128>), complex128);
33 QrOp<complex128>);
Dqr_op_complex64.cc20 REGISTER_LINALG_OP("Qr", (QrOp<complex64>), complex64);
Dqr_op_impl.h54 class QrOp : public LinearAlgebraOp<Scalar> {
58 explicit QrOp(OpKernelConstruction* context) : Base(context) { in QrOp() function
125 TF_DISALLOW_COPY_AND_ASSIGN(QrOp);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc458 static LogicalResult Verify(QrOp op) { in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc5586 class ConvertQrOp : public OpRewritePattern<TF::QrOp> {
5590 LogicalResult matchAndRewrite(TF::QrOp op, in matchAndRewrite()