experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Print out shapes' layouts in local_client's "shape/layout doesn't match" error.

Browse Source 
Previously we'd say "shape/layout doesn't match" and then print out the shapes
without their layouts!

PiperOrigin-RevId: 224105237
This commit is contained in:
Justin Lebar 2018-12-04 23:17:30 -08:00 committed by TensorFlower Gardener
parent 2114ed7447
commit 7c7ed2b64c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/client/local_client.cc
View File

@ -71,9 +71,9 @@ Status LocalExecutable::ValidateExecutionOptions(
"parameter "
"%d: want %s, got %s",
i,
ShapeUtil::HumanString(
ShapeUtil::HumanStringWithLayout(
computation_layout.parameter_layout(i).shape()),
ShapeUtil::HumanString(arguments[i]->on_host_shape()));
ShapeUtil::HumanStringWithLayout(arguments[i]->on_host_shape()));
}
}