diff options
author | Chris Leary <leary@google.com> | 2018-01-10 18:15:55 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-10 18:19:20 -0800 |
commit | 2b520270bb302998852377ab8fbe2eaf5ce377a9 (patch) | |
tree | 7ee61081638f95e4f3807c8e1ab9e750c50a715c /tensorflow/compiler/xla/client | |
parent | 964330c41e0435ea9a1122a64305ca91c82d1367 (diff) |
[XLA] More SWIG error propagation and transfers from outfeed.
PiperOrigin-RevId: 181547286
Diffstat (limited to 'tensorflow/compiler/xla/client')
-rw-r--r-- | tensorflow/compiler/xla/client/computation_builder.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/client/computation_builder.h b/tensorflow/compiler/xla/client/computation_builder.h index f11d769746..18ab24d9e6 100644 --- a/tensorflow/compiler/xla/client/computation_builder.h +++ b/tensorflow/compiler/xla/client/computation_builder.h @@ -422,8 +422,12 @@ class ComputationBuilder { // Enqueues an outfeed instruction onto the computation. This instruction // generates outgoing data transfers for the given data. - void Outfeed(const ComputationDataHandle& operand, const Shape& shape, - const string& outfeed_config); + // + // shape_with_layout communicates the laid out shape that we want to outfeed + // -- if !ShapeUtil::Compatible(GetShape(operand), shape_with_layout) an error + // will occur. + void Outfeed(const ComputationDataHandle& operand, + const Shape& shape_with_layout, const string& outfeed_config); // Enqueues a call instruction onto the computation. ComputationDataHandle Call( |