| /external/tensorflow/tensorflow/core/framework/ |
| D | log_memory.cc | 22 const string LogMemory::kLogMemoryLabel = "__LOG_MEMORY__";
24 bool LogMemory::IsEnabled() { return VLOG_IS_ON(2); } in IsEnabled()
34 LOG(INFO) << LogMemory::kLogMemoryLabel << " " << type_name << " { " in OutputToLog()
40 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep()
47 void LogMemory::RecordTensorAllocation(const string& kernel_name, in RecordTensorAllocation()
57 void LogMemory::RecordTensorDeallocation(const int64 allocation_id, in RecordTensorDeallocation()
65 void LogMemory::RecordTensorOutput(const string& kernel_name, in RecordTensorOutput()
76 void LogMemory::RecordRawAllocation(const string& operation, in RecordRawAllocation()
89 void LogMemory::RecordRawDeallocation(const string& operation, in RecordRawDeallocation()
|
| D | log_memory.h | 34 class LogMemory {
|
| D | op_kernel.cc | 271 if (LogMemory::IsEnabled()) { in allocate_temp()
272 LogMemory::RecordTensorAllocation( in allocate_temp()
273 def().name(), LogMemory::OP_KERNEL_CONSTRUCTION_STEP_ID, new_temp); in allocate_temp()
296 if (LogMemory::IsEnabled()) { in allocate_temp()
297 LogMemory::RecordTensorAllocation( in allocate_temp()
298 def().name(), LogMemory::OP_KERNEL_CONSTRUCTION_STEP_ID, new_temp); in allocate_temp()
723 LogMemory::RecordTensorAllocation(params_->op_kernel->name(), in allocate_tensor()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 45 if (tensorflow::LogMemory::IsEnabled() && data() != nullptr) { in LogAllocation()
46 tensorflow::LogMemory::RecordRawAllocation( in LogAllocation()
48 tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, size(), in LogAllocation()
53 if (tensorflow::LogMemory::IsEnabled() && data() != nullptr) { in LogDeallocation()
54 tensorflow::LogMemory::RecordRawDeallocation( in LogDeallocation()
56 tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, data(), in LogDeallocation()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 41 if (LogMemory::IsEnabled() && data != nullptr) { in allocate_tensor()
42 LogMemory::RecordRawAllocation( in allocate_tensor()
43 operation, LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, len, data, in allocate_tensor()
60 if (LogMemory::IsEnabled() && data != nullptr) { in deallocate_buffer()
61 LogMemory::RecordRawDeallocation( in deallocate_buffer()
62 "TensorFlow C Api", LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, data, in deallocate_buffer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 154 if (LogMemory::IsEnabled()) { in SetProtoFromGPU()
155 LogMemory::RecordRawAllocation("SetProtoFromGPU", in SetProtoFromGPU()
156 LogMemory::PROTO_BUFFER_STEP_ID, in SetProtoFromGPU()
175 if (LogMemory::IsEnabled()) { in SetProtoFromGPU()
176 LogMemory::RecordRawDeallocation("SetProtoFromGPU", in SetProtoFromGPU()
177 LogMemory::PROTO_BUFFER_STEP_ID, in SetProtoFromGPU()
|
| D | gpu_device.cc | 125 if (LogMemory::IsEnabled()) { in Reinitialize()
158 if (LogMemory::IsEnabled() && ret != nullptr) { in allocate()
159 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
165 if (LogMemory::IsEnabled() && buffer != nullptr) { in deallocate()
166 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_, in deallocate()
208 if (LogMemory::IsEnabled()) { in asyncFree()
209 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_, in asyncFree()
|
| D | gpu_process_state.cc | 316 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetGpuHostAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 120 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetCPUAllocator()
|
| D | graph_runner.cc | 178 args.step_id = LogMemory::CONSTANT_FOLDING_STEP_ID; in Run()
|
| D | direct_session.cc | 889 if (LogMemory::IsEnabled()) { in Run()
890 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()
1001 if (LogMemory::IsEnabled()) { in PRunSetup()
1002 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()
1469 if (LogMemory::IsEnabled() || run_state_args->is_partial_run) { in GetOrCreateExecutors()
1987 if (LogMemory::IsEnabled()) { in RunCallable()
1988 LogMemory::RecordStep(step_id, run_state_args.handle); in RunCallable()
|
| D | executor.cc | 395 log_memory_(LogMemory::IsEnabled()), in ExecutorState()
1032 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()
1041 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 538 if (LogMemory::IsEnabled()) { in StartParallelExecutors()
539 LogMemory::RecordStep(args.step_id, handle); in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 56 kernel->reset(new KernelAndDeviceOp(ctx.GetRendezvous(), ctx.LogMemory(), flr, in CreateUncachedKernelAndDeviceOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 269 bool LogMemory() const { return log_memory_; } in LogMemory() function
|
| D | execute.cc | 537 ctx.GetRendezvous(), ctx.LogMemory(), flr, runner, in GetOrCreateKernelAndDevice()
|
| D | context.cc | 95 log_memory_(LogMemory::IsEnabled()),
|