Searched refs:DiagPartOp (Results 1 – 2 of 2) sorted by relevance
73 class DiagPartOp : public OpKernel { class75 explicit DiagPartOp(OpKernelConstruction* context) : OpKernel(context) {} in DiagPartOp() function in tensorflow::DiagPartOp186 DiagPartOp<CPUDevice, T>)235 DiagPartOp<GPUDevice, T>)
121 class DiagPartOp : public XlaOpKernel { class123 explicit DiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in DiagPartOp() function in tensorflow::__anon89a1e8320111::DiagPartOp161 REGISTER_XLA_OP(Name("DiagPart"), DiagPartOp);