Searched refs:MatrixDiagPartOp (Results 1 – 2 of 2) sorted by relevance
44 class MatrixDiagPartOp : public OpKernel { class46 explicit MatrixDiagPartOp(OpKernelConstruction* context) in MatrixDiagPartOp() function in tensorflow::MatrixDiagPartOp80 TF_DISALLOW_COPY_AND_ASSIGN(MatrixDiagPartOp);125 MatrixDiagPartOp<CPUDevice, type>);138 MatrixDiagPartOp<CPUDevice, type>);203 MatrixDiagPartOp<GPUDevice, type>);219 MatrixDiagPartOp<GPUDevice, type>);
192 class MatrixDiagPartOp : public XlaOpKernel { class194 explicit MatrixDiagPartOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in MatrixDiagPartOp() function in tensorflow::__anon89a1e8320111::MatrixDiagPartOp209 REGISTER_XLA_OP(Name("MatrixDiagPart"), MatrixDiagPartOp);