Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc75 class DiagOp : public XlaOpKernel { class
77 explicit DiagOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagOp() function in tensorflow::__anonc136a9900111::DiagOp
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc43 class DiagOp : public OpKernel { class
45 explicit DiagOp(OpKernelConstruction* context) : OpKernel(context) {} in DiagOp() function in tensorflow::DiagOp