diff options
author | 2018-07-10 10:22:11 -0700 | |
---|---|---|
committer | 2018-07-10 10:25:50 -0700 | |
commit | a6df304e5ec4b0ad28884025558ecda414b9df07 (patch) | |
tree | 3727fe7253c5eb8b6fb2f1e1cf02705398828a2b | |
parent | 7aeb0807b6293067534655709977861b2fec529b (diff) |
Fix possible segfault in shape function for SymbolicGradient
PiperOrigin-RevId: 203969340
-rw-r--r-- | tensorflow/core/ops/functional_ops.cc | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/core/ops/functional_ops.cc b/tensorflow/core/ops/functional_ops.cc index b49ad8e387..5f262db2ce 100644 --- a/tensorflow/core/ops/functional_ops.cc +++ b/tensorflow/core/ops/functional_ops.cc @@ -40,7 +40,11 @@ REGISTER_OP("SymbolicGradient") if (types[i] == DT_RESOURCE) { const std::vector<shape_inference::ShapeAndType>* handle_type = c->input_handle_shapes_and_types(i); - c->set_output(i, handle_type->at(0).shape); + if (handle_type != nullptr) { + c->set_output(i, handle_type->at(0).shape); + } else { + c->set_output(i, c->UnknownShape()); + } } else { c->set_output(i, c->input(i)); } |