Searched refs:MatrixDiagOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 151 class MatrixDiagOp : public OpKernel { class
153 explicit MatrixDiagOp(OpKernelConstruction* context) : OpKernel(context) { in MatrixDiagOp() function in tensorflow::MatrixDiagOp
266 TF_DISALLOW_COPY_AND_ASSIGN(MatrixDiagOp);
272 MatrixDiagOp<CPUDevice, type>); \
275 MatrixDiagOp<CPUDevice, type>); \
278 MatrixDiagOp<CPUDevice, type>); \
297 MatrixDiagOp<CPUDevice, type>); \
480 MatrixDiagOp<GPUDevice, type>); \
488 MatrixDiagOp<GPUDevice, type>); \
496 MatrixDiagOp<GPUDevice, type>); \
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 230 class MatrixDiagOp : public XlaOpKernel { class
232 explicit MatrixDiagOp(OpKernelConstruction* context) : XlaOpKernel(context) { in MatrixDiagOp() function in tensorflow::MatrixDiagOp
330 REGISTER_XLA_OP(Name("MatrixDiag"), MatrixDiagOp);
336 MatrixDiagOp);
342 MatrixDiagOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 427 rewriter->replaceOpWithNewOp<MatrixDiagOp>(op, output_type, input); in ConvertTFMatrixDiagV2orV3()
|