[MLIR] Add e2e test for unranked unary TF op, lowered and run with CPU runner.

PiperOrigin-RevId: 325665428
2020-08-09 02:36:32 -07:00 · 2020-08-09 02:36:32 -07:00 · 843af36e05
parent 417e9e9c34
commit 843af36e05
1 changed files with 1 additions and 1 deletions
--- a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
+++ b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
@ -170,7 +170,7 @@ struct TransformUnrankedHloPass
    PopulateTransformUnrankedHloPatterns(&ctx, &patterns);

    // Apply transformation.
-    if (failed(applyFullConversion(getFunction(), target, patterns)))
+    if (failed(applyPartialConversion(getFunction(), target, patterns)))
      return signalPassFailure();
  }
 };