Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc121 class DiagPartOp : public XlaOpKernel { class
123 explicit DiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagPartOp() function in tensorflow::__anon89a1e8320111::DiagPartOp
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc73 class DiagPartOp : public OpKernel { class
75 explicit DiagPartOp(OpKernelConstruction* context) : OpKernel(context) {} in DiagPartOp() function in tensorflow::DiagPartOp