diff options
Diffstat (limited to 'tensorflow/core/kernels/sdca_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/sdca_ops.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/sdca_ops.cc b/tensorflow/core/kernels/sdca_ops.cc index d7365c6560..eb679cf2ae 100644 --- a/tensorflow/core/kernels/sdca_ops.cc +++ b/tensorflow/core/kernels/sdca_ops.cc @@ -142,7 +142,8 @@ void DoCompute(const ComputeOptions& options, OpKernelContext* const context) { Tensor mutable_example_state_data_t(*example_state_data_t); auto example_state_data = mutable_example_state_data_t.matrix<float>(); - context->set_output("out_example_state_data", mutable_example_state_data_t); + OP_REQUIRES_OK(context, context->set_output("out_example_state_data", + mutable_example_state_data_t)); if (options.adaptative) { OP_REQUIRES_OK(context, |