Searched refs:log_memory (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 195 tensorflow::Rendezvous* rendezvous, bool log_memory, in KernelAndDeviceOp() argument
203 log_memory_(log_memory) {} in KernelAndDeviceOp()
|
| D | kernel_and_device.cc | 285 params.log_memory = log_memory_; in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 127 "log_memory.proto",
196 "log_memory.h",
306 "log_memory.cc",
307 "log_memory.h",
781 "log_memory.cc",
790 "log_memory.h",
1086 "log_memory.h",
1404 srcs = ["log_memory.proto"],
|
| D | op_kernel.h | 594 bool log_memory = false; member
|
| D | op_kernel.cc | 699 if (params_->log_memory) { in allocate_tensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 306 params.log_memory = false; // TODO(mrry): Too severe? in Run()
|
| D | executor.cc | 703 params.log_memory = log_memory_; in Process()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 283 params->log_memory = false; in PrepOpKernel()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 210 "//tensorflow/core/framework:log_memory.proto",
440 "//tensorflow/core/framework:log_memory.h",
|