Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_set_diag_op.cc44 class MatrixSetDiagOp : public OpKernel { class
46 explicit MatrixSetDiagOp(OpKernelConstruction* context) : OpKernel(context) { in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmatrix_diag_ops.cc457 class MatrixSetDiagOp : public XlaOpKernel { class
459 explicit MatrixSetDiagOp(OpKernelConstruction* context) in MatrixSetDiagOp() function in tensorflow::MatrixSetDiagOp