diff options
author | Thomas Joerg <tjoerg@google.com> | 2018-09-21 09:10:17 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 09:13:56 -0700 |
commit | 5877baddc72e3f234f6e0a174447becd4cabc493 (patch) | |
tree | 5f9ea4d517ba58a93675d46f7a36ee103fdaabcc /tensorflow/compiler/jit | |
parent | b79b97cd23a7f8c308c9e8e6fcd425d6a8e9c243 (diff) |
[XLA] Dump the original, unclustered graph with --tf_xla_clustering_debug.
So far, just the clustered graph is dumped.
PiperOrigin-RevId: 213994376
Diffstat (limited to 'tensorflow/compiler/jit')
-rw-r--r-- | tensorflow/compiler/jit/mark_for_compilation_pass.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass.cc b/tensorflow/compiler/jit/mark_for_compilation_pass.cc index 1eaedbfbfb..133d982360 100644 --- a/tensorflow/compiler/jit/mark_for_compilation_pass.cc +++ b/tensorflow/compiler/jit/mark_for_compilation_pass.cc @@ -982,6 +982,11 @@ Status MarkForCompilationPass::RunImpl( // Names for each cluster. std::unordered_map<int, string> cluster_names; + if (flags->tf_xla_clustering_debug) { + dump_graph::DumpGraphToFile("before_mark_for_compilation", **options.graph, + options.flib_def); + } + // Mark clusters for compilation that: // * are placed on a device that requires compilation (an XlaDevice), // * are explicitly marked for compilation (_XlaCompile=true), or |