diff options
Diffstat (limited to 'tensorflow/core/kernels/function_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/function_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/function_ops.cc b/tensorflow/core/kernels/function_ops.cc index 58c4ed37c4..71f1b4e063 100644 --- a/tensorflow/core/kernels/function_ops.cc +++ b/tensorflow/core/kernels/function_ops.cc @@ -237,6 +237,8 @@ class SymbolicGradientOp : public AsyncOpKernel { FunctionLibraryRuntime::Options opts; opts.step_id = ctx->step_id(); + opts.rendezvous = ctx->rendezvous(); + opts.cancellation_manager = ctx->cancellation_manager(); opts.runner = ctx->runner(); std::vector<Tensor> args; args.reserve(ctx->num_inputs()); |