Searched refs:log_memory (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 201 tensorflow::Rendezvous* rendezvous, bool log_memory, in KernelAndDeviceOp() argument
209 log_memory_(log_memory) {} in KernelAndDeviceOp()
|
| D | kernel_and_device.cc | 300 params.log_memory = log_memory_; in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 129 "log_memory.proto",
200 "log_memory.h",
312 "log_memory.cc",
313 "log_memory.h",
803 "log_memory.cc",
812 "log_memory.h",
1125 "log_memory.h",
1495 srcs = ["log_memory.proto"],
|
| D | Android.bp | 83 "log_memory.cc",
|
| D | op_kernel.h | 607 bool log_memory = false; member
|
| D | op_kernel.cc | 766 if (params_->log_memory) { in allocate_tensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 312 params.log_memory = false; // TODO(mrry): Too severe? in Run()
|
| D | executor.cc | 734 params.log_memory = log_memory_; in ProcessInline()
|
| /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 | 218 "//tensorflow/core/framework:log_memory.proto",
452 "//tensorflow/core/framework:log_memory.h",
|