| commit | 202e329b58c94a4a59f2043011904951b9412240 | [log] [tgz] |
|---|---|---|
| author | Ruoxin Sang <rxsang@google.com> | Tue Aug 25 17:09:26 2020 -0700 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Tue Aug 25 17:24:12 2020 -0700 |
| tree | 7e0e7d0773c2465e8a8daa30c12f3105705e42c0 | |
| parent | da0d67335f40b24f057e1a8a19ffc64fd2237a4d [diff] |
Add more informative logging if `TF_RET_CHECK(arg_shape.handle_type != DT_INVALID)` fails. PiperOrigin-RevId: 328434176 Change-Id: Ic7c359c4eb4ff9a77a022a4a94402b1415ac88b7
diff --git a/tensorflow/core/tpu/graph_rewrite/distributed_tpu_rewrite_pass.cc b/tensorflow/core/tpu/graph_rewrite/distributed_tpu_rewrite_pass.cc index 2544e3f..cdf32c5 100644 --- a/tensorflow/core/tpu/graph_rewrite/distributed_tpu_rewrite_pass.cc +++ b/tensorflow/core/tpu/graph_rewrite/distributed_tpu_rewrite_pass.cc
@@ -1568,7 +1568,8 @@ arg_shape.shape = TensorShape(); // Variables are always scalars. arg_shape.handle_shape = info->handle_shape; arg_shape.handle_type = info->handle_type; - TF_RET_CHECK(arg_shape.handle_type != DT_INVALID); + TF_RET_CHECK(arg_shape.handle_type != DT_INVALID) + << " input edge: " << input_edges[edge_pos]->DebugString(); ++edge_pos; }