Searched refs:log_memory (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 193 tensorflow::Rendezvous* rendezvous, bool log_memory, in KernelAndDeviceOp() argument
201 log_memory_(log_memory) {} in KernelAndDeviceOp()
|
| D | kernel_and_device.cc | 278 params.log_memory = log_memory_; in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 128 "log_memory.proto",
198 "log_memory.h",
306 "log_memory.cc",
307 "log_memory.h",
747 "log_memory.cc",
756 "log_memory.h",
1044 "log_memory.h",
1386 srcs = ["log_memory.proto"],
|
| D | op_kernel.h | 634 bool log_memory = false; member
|
| D | op_kernel.cc | 722 if (params_->log_memory) { in allocate_tensor()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 289 params.log_memory = false; // TODO(mrry): Too severe? in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 283 params->log_memory = false; in PrepOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 697 params.log_memory = log_memory_; in Process()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 202 "//tensorflow/core/framework:log_memory.proto",
431 "//tensorflow/core/framework:log_memory.h",
|