diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-25 19:17:40 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-25 19:21:43 -0700 |
| commit | 689cbda96444511bd37a01b125791c45a093bec3 (patch) | |
| tree | 5fe2bd2bed50c943d95a9fa0a44d35cf2c167fa0 /tensorflow/core/kernels/function_ops.cc | |
| parent | 42eba6a7046d232de397ea2d0e627e91f095bc4c (diff) | |
Plumbs rendezvous through function calls.
PiperOrigin-RevId: 163157033
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()); |