experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add vlog for when unsupported ops are marked for outside compilation.

Browse Source 
PiperOrigin-RevId: 348050595
Change-Id: I72221b6db94f04a3cf363bdc0754e8eb5e06d658
This commit is contained in:
Ken Franko 2020-12-17 10:44:34 -08:00 committed by TensorFlower Gardener
parent 1ea36b6b53
commit 911a08098a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/tensorflow/transforms/mark_ops_for_outside_compilation.cc
View File

@ -198,6 +198,9 @@ LogicalResult MarkUncompilableOps(
int outside_compiled_cluster_counter = 0; int outside_compiled_cluster_counter = 0;
block->walk([&](Operation* op) { block->walk([&](Operation* op) {
if (!IsSupportedOp(*op, supported_ops, tf_dialect)) { 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( op->setAttr(
kXlaOutsideCompilationAttr, kXlaOutsideCompilationAttr,
StringAttr::get( StringAttr::get(