From 911a08098a97ad9d296e99dc2d6cbae296344b2f Mon Sep 17 00:00:00 2001
From: Ken Franko <kfranko@google.com>
Date: Thu, 17 Dec 2020 10:44:34 -0800
Subject: [PATCH] Add vlog for when unsupported ops are marked for outside
 compilation.

PiperOrigin-RevId: 348050595
Change-Id: I72221b6db94f04a3cf363bdc0754e8eb5e06d658
---
 .../tensorflow/transforms/mark_ops_for_outside_compilation.cc  | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/mark_ops_for_outside_compilation.cc b/tensorflow/compiler/mlir/tensorflow/transforms/mark_ops_for_outside_compilation.cc
index 9570915b931..f9a85b71808 100644
--- a/tensorflow/compiler/mlir/tensorflow/transforms/mark_ops_for_outside_compilation.cc
+++ b/tensorflow/compiler/mlir/tensorflow/transforms/mark_ops_for_outside_compilation.cc
@@ -198,6 +198,9 @@ LogicalResult MarkUncompilableOps(
   int outside_compiled_cluster_counter = 0;
   block->walk([&](Operation* op) {
     if (!IsSupportedOp(*op, supported_ops, tf_dialect)) {
+      VLOG(3) << "Cloud TPU: Op " << op->getName().getStringRef().str()
+              << " isn't compilable, adding outside_compilation attr. "
+                 "This op will automatically be placed on CPU.";
       op->setAttr(
           kXlaOutsideCompilationAttr,
           StringAttr::get(