diff options
| author |  Derek Murray <mrry@google.com> | 2017-12-13 08:56:20 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-13 09:00:51 -0800 |
| commit | 185c593cb71cb6d8116ba05c97e9385642648f1b (patch) | |
| tree | 8853277b53c58d69e93735be50e75f4f5afa9516 /tensorflow/c/c_api_function.cc | |
| parent | 2b1b7dffcd2c76876efdbcfc431424e259da3bf4 (diff) | |
Automated g4 rollback of changelist 178759398
PiperOrigin-RevId: 178909147
Diffstat (limited to 'tensorflow/c/c_api_function.cc')
| -rw-r--r-- | tensorflow/c/c_api_function.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_function.cc b/tensorflow/c/c_api_function.cc index dcb818b88b..b9312c2974 100644 --- a/tensorflow/c/c_api_function.cc +++ b/tensorflow/c/c_api_function.cc @@ -226,6 +226,11 @@ Status FillFunctionBody( } node_def->add_input(strings::StrCat("^", normalized)); } + + // A function is stateful if any of its nodes are stateful. + if (node->op_def().is_stateful()) { + fdef->mutable_signature()->set_is_stateful(true); + } } return Status::OK(); } |