diff options
| author |  Jacques Pienaar <jpienaar@google.com> | 2018-03-09 11:37:04 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-09 11:41:08 -0800 |
| commit | 61a744fffbcc68e453aafc6eaa2c7ff2318a3584 (patch) | |
| tree | bca63e09f31f998c6a7e7227baf3955c9d85712c /tensorflow/compiler/xla/service/shape_inference.cc | |
| parent | 46c2d1a6c4c65883fa4a37f3737f1cdc0eebceef (diff) | |
Add more debugging output, filtering by int_type in XLA test, more tests.
PiperOrigin-RevId: 188513895
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.cc b/tensorflow/compiler/xla/service/shape_inference.cc index 2ff7ae97b7..74f744a62b 100644 --- a/tensorflow/compiler/xla/service/shape_inference.cc +++ b/tensorflow/compiler/xla/service/shape_inference.cc @@ -193,7 +193,10 @@ tensorflow::Status VerifyReducerShape(const ProgramShape& reducer_shape, const Shape& accumulator_shape = reducer_shape.result(); if (ShapeUtil::Rank(accumulator_shape) != 0) { - return InvalidArgument("Reduction function must have rank 0."); + return InvalidArgument( + "Reduction function must have rank 0 (rank %lld reduction function " + "given).", + ShapeUtil::Rank(accumulator_shape)); } // Check that the accumulator can be passed in as the first argument. |