diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-13 11:37:37 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-13 11:45:43 -0800 |
| commit | 3db96abfc5432c190d3afa62ebfad3c1d82cd818 (patch) | |
| tree | a5926f485e408ad968a77f1e540dbcdeff2f23bb /tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc | |
| parent | 58f7858601b72aa3c5854571f2152b91d1795e29 (diff) | |
Allow assigning colors based on HLO sharding information, when generating Graphviz HLO graphs via a new --xla_hlo_graph_sharding_color option.
When generating TF graphs, a new --xla_hlo_tfgraph_device_scopes option allows to prefix the instructions names with a device scope. This help the TF graph viewer to better isolate the parts of the graph which are targeted to different devices, and allow rendering of graphs which would not be able to due to size.
Changed TF/XLA broadcast lowering to propagate the request metadata into the HLO broadcast instructions.
PiperOrigin-RevId: 175563052
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc b/tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc index 7b0f937f38..8e1531c87f 100644 --- a/tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc +++ b/tensorflow/compiler/xla/service/hlo_graph_dumper_test.cc @@ -45,7 +45,7 @@ class DotRenderer : public hlo_graph_dumper::GraphRendererInterface { string last_graph_; }; -XLA_REGISTER_GRAPH_RENDERER(DotRenderer, std::numeric_limits<int>::max()); +XLA_REGISTER_GRAPH_RENDERER(DotRenderer); TEST(HloGraphDumperTest, NestedFusion) { HloComputation::Builder b("b"); |