diff options
author | 2018-05-01 09:07:57 -0700 | |
---|---|---|
committer | 2018-05-01 09:10:07 -0700 | |
commit | a82e0e7922d6dc657b42ef2b3a7a1a52194454c8 (patch) | |
tree | f9bec0d15c8b579dd06e68c2e9be0b310fb1a9af /tensorflow/compiler/xla/service/hlo_graph_dumper.cc | |
parent | 10337c91efe7e3975134a7b09ea598e85877c1b0 (diff) |
Fix crash in HloGraphDumper where it crashes on tuple shaped constants
The problem is that it tries to use a special logic for 0 element constants
but the logic used to check the number of elements only supports array shapes.
PiperOrigin-RevId: 194945246
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_graph_dumper.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_graph_dumper.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_graph_dumper.cc b/tensorflow/compiler/xla/service/hlo_graph_dumper.cc index 516e14b464..bb4db89f0a 100644 --- a/tensorflow/compiler/xla/service/hlo_graph_dumper.cc +++ b/tensorflow/compiler/xla/service/hlo_graph_dumper.cc @@ -804,7 +804,7 @@ string HloDotDumper::GetInstructionNodeInlinedOperands( // "{} (f32[42, 0, 10])". The alternative, calling Literal::ToString(), // enumerates all of its empty dimensions (e.g. "{ { {}, {} }, ..."), which // is just noise. - if (ShapeUtil::HasZeroElements(shape)) { + if (!ShapeUtil::IsTuple(shape) && ShapeUtil::HasZeroElements(shape)) { return Printf("{} (%s)", ShapeUtil::HumanString(constant->shape())); } |