Searched defs:MatrixDiagPartOp (Results 1 – 2 of 2) sorted by relevance
344 class MatrixDiagPartOp : public XlaOpKernel { class346 explicit MatrixDiagPartOp(OpKernelConstruction* context) in MatrixDiagPartOp() function in tensorflow::MatrixDiagPartOp
47 class MatrixDiagPartOp : public OpKernel { class49 explicit MatrixDiagPartOp(OpKernelConstruction* context) : OpKernel(context) { in MatrixDiagPartOp() function in tensorflow::MatrixDiagPartOp