aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/hlo_graph_dumper.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-02-16 14:34:16 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-02-16 14:37:59 -0800
commit13df417665f216bfb527440f1fd8f04958000ec5 (patch)
tree7e3ed509f8edb484926d3c7ef9b523179ca7061f /tensorflow/compiler/xla/service/hlo_graph_dumper.cc
parentdd2efdf8cf0346e49592d7a339f09b1862a256c6 (diff)
[TF:XLA] Adds HostCompute HLO - a pseudo-op to represent host-side computation.
PiperOrigin-RevId: 186047964
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_graph_dumper.cc')
-rw-r--r--tensorflow/compiler/xla/service/hlo_graph_dumper.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_graph_dumper.cc b/tensorflow/compiler/xla/service/hlo_graph_dumper.cc
index 44fcd36370..9b0e2fd7d6 100644
--- a/tensorflow/compiler/xla/service/hlo_graph_dumper.cc
+++ b/tensorflow/compiler/xla/service/hlo_graph_dumper.cc
@@ -988,6 +988,7 @@ ColorScheme HloDotDumper::GetInstructionColor(const HloInstruction* instr) {
case HloOpcode::kCall:
case HloOpcode::kConditional:
case HloOpcode::kCustomCall:
+ case HloOpcode::kHostCompute:
case HloOpcode::kWhile:
return kDarkGreen;
case HloOpcode::kConstant: