diff options
Diffstat (limited to 'tensorflow/core/common_runtime/eager/execute.cc')
-rw-r--r-- | tensorflow/core/common_runtime/eager/execute.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/common_runtime/eager/execute.cc b/tensorflow/core/common_runtime/eager/execute.cc index 5b3a64ba98..1da1326a9a 100644 --- a/tensorflow/core/common_runtime/eager/execute.cc +++ b/tensorflow/core/common_runtime/eager/execute.cc @@ -296,7 +296,7 @@ Status EagerLocalExecute(EagerOperation* op, LOG(INFO) << "Executing op " << ndef.op() << " in device " << device->name(); } - kernel = new KernelAndDevice(ctx->GetRendezvous()); + kernel = new KernelAndDevice(ctx->GetRendezvous(), ctx->LogMemory()); auto* flr = ctx->func_lib(device); if (flr == nullptr) { |