Searched defs:MatrixSetDiagOp (Results 1 – 2 of 2) sorted by relevance
44 class MatrixSetDiagOp : public OpKernel { class46 explicit MatrixSetDiagOp(OpKernelConstruction* context) : OpKernel(context) { in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp
454 class MatrixSetDiagOp : public XlaOpKernel { class456 explicit MatrixSetDiagOp(OpKernelConstruction* context) in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp