diff options
author | Peter Hawkins <phawkins@google.com> | 2017-08-05 10:55:39 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-05 10:59:53 -0700 |
commit | 5d52208e6e320d91c00154f1e394b9f0ee4b404e (patch) | |
tree | a51c9c7d7ab241bf135a36fa9890822b45ecb882 /tensorflow/compiler/xla/layout_util.cc | |
parent | c62eaccec8fe303a70246085160f803ffe0f2029 (diff) |
[TF:XLA] Fixes to the "evaluator" plugin.
* Mark the evaluator plugin as alwayslink so it doesn't get stripped out by the linker.
* Add a generic LayoutAssignment pass to the pass pipeline; otherwise the entry computation has no layout and Service::Execute CHECK-fails in the AllocationTracker.
* Register the default computation placer for the evaluator backend.
* Add an replay_computation_hlo_evaluator binary that can replay computation snapshots via the HLO evaluator.
PiperOrigin-RevId: 164364780
Diffstat (limited to 'tensorflow/compiler/xla/layout_util.cc')
-rw-r--r-- | tensorflow/compiler/xla/layout_util.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/layout_util.cc b/tensorflow/compiler/xla/layout_util.cc index 35a563bf22..6271b59a5b 100644 --- a/tensorflow/compiler/xla/layout_util.cc +++ b/tensorflow/compiler/xla/layout_util.cc @@ -133,7 +133,8 @@ Layout CreateDefaultLayoutForRank(int64 rank) { } else { // Array shape. if (!shape.has_layout()) { - return InvalidArgument("shape does not have a layout"); + return InvalidArgument("shape %s does not have a layout", + ShapeUtil::HumanString(shape).c_str()); } return ValidateLayoutForShape(shape.layout(), shape); } |