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.cc76 class DiagOp : public XlaOpKernel { class
78 explicit DiagOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagOp() function in tensorflow::__anonad1ac0750111::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