diff options
| author |  Ayush Dubey <ayushd@google.com> | 2018-08-30 12:07:35 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 12:13:36 -0700 |
| commit | c44566cdd632e4a9b030244cbc36965cb0ee21c0 (patch) | |
| tree | 85e67b192bbbbbfc705d04cc9246476fd65b65de /tensorflow/core/common_runtime/function.cc | |
| parent | 446a7b543a6f805bbfbc84b35c562b0282ceacf2 (diff) | |
Set collective_executor in function CallOp from OpKernelContext.
Without this change, CollectiveOps in functions cannot execute.
PiperOrigin-RevId: 210955255
Diffstat (limited to 'tensorflow/core/common_runtime/function.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/function.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/function.cc b/tensorflow/core/common_runtime/function.cc index 2ef7d5d5ca..46bb8d92f8 100644 --- a/tensorflow/core/common_runtime/function.cc +++ b/tensorflow/core/common_runtime/function.cc @@ -310,6 +310,7 @@ class CallOp : public AsyncOpKernel { opts.step_container = ctx->step_container(); opts.stats_collector = ctx->stats_collector(); opts.runner = ctx->runner(); + opts.collective_executor = ctx->collective_executor(); std::vector<Tensor> args; args.reserve(ctx->num_inputs()); for (int i = 0; i < ctx->num_inputs(); ++i) { |