diff options
author | avijit-nervana <avijit.chakraborty@intel.com> | 2018-09-14 09:21:08 -0700 |
---|---|---|
committer | avijit-nervana <avijit.chakraborty@intel.com> | 2018-09-14 09:21:08 -0700 |
commit | 41aaed7751690b0b3137dad2620656a698b3ceae (patch) | |
tree | 00fc1a7f6be0c3968f3e674a65ca4907110ddf2d /tensorflow/core/common_runtime/eager/kernel_and_device.cc | |
parent | c26c5e1217944448f1f4c2b97626fc4d7d6406d3 (diff) | |
parent | 95338704198205c1bdec1e344e103f1daf05df68 (diff) |
Merge branch 'master' into avijit/add-cpu-backend
Diffstat (limited to 'tensorflow/core/common_runtime/eager/kernel_and_device.cc')
-rw-r--r-- | tensorflow/core/common_runtime/eager/kernel_and_device.cc | 16 |
1 files changed, 1 insertions, 15 deletions
diff --git a/tensorflow/core/common_runtime/eager/kernel_and_device.cc b/tensorflow/core/common_runtime/eager/kernel_and_device.cc index 3d61ff4dc2..83d8425477 100644 --- a/tensorflow/core/common_runtime/eager/kernel_and_device.cc +++ b/tensorflow/core/common_runtime/eager/kernel_and_device.cc @@ -32,21 +32,6 @@ limitations under the License. namespace tensorflow { // static -Status KernelAndDevice::InitOp(Device* device, const NodeDef& ndef, - KernelAndDevice* out) { - OpKernel* k = nullptr; - Status s = CreateOpKernel(device->device_type().c_str(), device, - device->GetAllocator(AllocatorAttributes()), - nullptr, ndef, TF_GRAPH_DEF_VERSION, &k); - out->device_ = device; - out->kernel_.reset(k); - out->flib_ = nullptr; - out->runner_ = nullptr; - out->default_runner_ = [](std::function<void()> f) { f(); }; - return s; -} - -// static Status KernelAndDevice::Init(const NodeDef& ndef, FunctionLibraryRuntime* flib, std::function<void(std::function<void()>)>* runner, KernelAndDevice* out) { @@ -95,6 +80,7 @@ Status KernelAndDevice::Run(ScopedStepContainer* step_container, params.slice_reader_cache = &slice_reader_cache_; params.rendezvous = rendez_; params.cancellation_manager = &cm_; + params.log_memory = log_memory_; if (stats != nullptr) { params.track_allocations = true; } |