Home
last modified time | relevance | path

Searched defs:MatrixDiagOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatrix_diag_op.cc84 class MatrixDiagOp : public OpKernel { class
86 explicit MatrixDiagOp(OpKernelConstruction* context) : OpKernel(context) {} in MatrixDiagOp() function in tensorflow::MatrixDiagOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc163 class MatrixDiagOp : public XlaOpKernel { class
165 explicit MatrixDiagOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in MatrixDiagOp() function in tensorflow::__anon89a1e8320111::MatrixDiagOp