Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmatrix_diag_op.cc44 class MatrixDiagPartOp : public OpKernel { class
46 explicit MatrixDiagPartOp(OpKernelConstruction* context) in MatrixDiagPartOp() function in tensorflow::MatrixDiagPartOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc192 class MatrixDiagPartOp : public XlaOpKernel { class
194 explicit MatrixDiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in MatrixDiagPartOp() function in tensorflow::__anon89a1e8320111::MatrixDiagPartOp