OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

In the overloaded HloVerifier::CheckShape, include the failing instruction in

Browse Source 
the error message.

PiperOrigin-RevId: 173965368
This commit is contained in:
A. Unique TensorFlower
2017-10-30 16:09:32 -07:00
committed by TensorFlower Gardener
parent 302ab0ff76
commit f9a673cb71
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/xla/service/hlo_verifier.cc
View File

@@ -359,7 +359,10 @@ class ShapeVerifier : public DfsHloVisitor {
Status CheckShape(const HloInstruction* instruction,
const StatusOr<Shape>& expected_shape_status) {
if (!expected_shape_status.ok()) {
return expected_shape_status.status();
Status s = expected_shape_status.status();
tensorflow::errors::AppendToMessage(&s, ", for instruction ",
instruction->ToString());
return s;
}
return CheckShape(instruction, expected_shape_status.ValueOrDie());
}