- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for MatrixDiagV2OrV3Op (0.11 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// to call `rewriter::replaceOpWihNewOp`, which is not a const member function. template <typename MatrixDiagV2OrV3Op> bool ConvertTFMatrixDiagV2orV3(Operation* op, PatternRewriter* rewriter) { auto tf_matrix_diag_v2_or_v3_op = cast<MatrixDiagV2OrV3Op>(op); if (tf_matrix_diag_v2_or_v3_op.getNumOperands() != 5) return false; auto input = tf_matrix_diag_v2_or_v3_op.getDiagonal();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (1)