Searched refs:log_memory (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 105 "log_memory.proto",
168 "log_memory.h",
268 "log_memory.cc",
269 "log_memory.h",
678 "log_memory.cc",
687 "log_memory.h",
923 "log_memory.h",
1136 srcs = ["log_memory.proto"],
|
| D | op_kernel.h | 664 bool log_memory = false; member
|
| D | op_kernel.cc | 764 if (params_->log_memory) { in allocate_tensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 171 tensorflow::Rendezvous* rendez, bool log_memory, in KernelAndDeviceOp() argument
179 log_memory_(log_memory), in KernelAndDeviceOp()
|
| D | kernel_and_device.cc | 271 params.log_memory = log_memory_; in Run()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 244 params.log_memory = false; // TODO(mrry): Too severe? in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 428 params->log_memory = false; in PrepOpKernel()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 221 "//tensorflow/core/framework:log_memory.proto",
470 "//tensorflow/core/framework:log_memory.h",
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 1711 params.log_memory = log_memory_; in Process()
|