| /external/tensorflow/tensorflow/core/framework/ |
| D | log_memory.cc | 23 const string LogMemory::kLogMemoryLabel = "__LOG_MEMORY__";
25 bool LogMemory::IsEnabled() { return VLOG_IS_ON(1); } in IsEnabled()
35 LOG(INFO) << LogMemory::kLogMemoryLabel << " " << type_name << " { " in OutputToLog()
41 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep()
48 void LogMemory::RecordTensorAllocation(const string& kernel_name, in RecordTensorAllocation()
58 void LogMemory::RecordTensorDeallocation(const int64 allocation_id, in RecordTensorDeallocation()
66 void LogMemory::RecordTensorOutput(const string& kernel_name, in RecordTensorOutput()
77 void LogMemory::RecordRawAllocation(const string& operation, in RecordRawAllocation()
90 void LogMemory::RecordRawDeallocation(const string& operation, in RecordRawDeallocation()
|
| D | tensor.cc | 97 LogMemory::RecordTensorDeallocation(alloc_->AllocationId(data()), in RecordDeallocation()
456 if (LogMemory::IsEnabled()) { in ~Buffer()
740 if (buf_ != nullptr && buf_->data() != nullptr && LogMemory::IsEnabled()) { in Tensor()
741 LogMemory::RecordTensorAllocation("Unknown", LogMemory::UNKNOWN_STEP_ID, in Tensor()
755 buf_->data() != nullptr && LogMemory::IsEnabled()) { in Tensor()
756 LogMemory::RecordTensorAllocation("Unknown (with attributes)", in Tensor()
757 LogMemory::UNKNOWN_STEP_ID, *this); in Tensor()
883 if (buf_ != nullptr && buf_->data() != nullptr && LogMemory::IsEnabled()) { in FromProto()
884 LogMemory::RecordTensorAllocation("Unknown (from Proto)", in FromProto()
885 LogMemory::UNKNOWN_STEP_ID, *this); in FromProto()
|
| D | log_memory.h | 34 class LogMemory {
|
| D | op_kernel.cc | 246 if (LogMemory::IsEnabled()) { in allocate_temp()
247 LogMemory::RecordTensorAllocation( in allocate_temp()
248 def_->name(), LogMemory::OP_KERNEL_CONSTRUCTION_STEP_ID, new_temp); in allocate_temp()
717 LogMemory::RecordTensorAllocation(params_->op_kernel->name(), in allocate_tensor()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 44 if (tensorflow::LogMemory::IsEnabled() && data() != nullptr) { in LogAllocation()
45 tensorflow::LogMemory::RecordRawAllocation( in LogAllocation()
47 tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, size(), in LogAllocation()
52 if (tensorflow::LogMemory::IsEnabled() && data() != nullptr) { in LogDeallocation()
53 tensorflow::LogMemory::RecordRawDeallocation( in LogDeallocation()
55 tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, data(), in LogDeallocation()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 155 if (LogMemory::IsEnabled()) { in SetProtoFromGPU()
156 LogMemory::RecordRawAllocation("SetProtoFromGPU", in SetProtoFromGPU()
157 LogMemory::PROTO_BUFFER_STEP_ID, in SetProtoFromGPU()
176 if (LogMemory::IsEnabled()) { in SetProtoFromGPU()
177 LogMemory::RecordRawDeallocation("SetProtoFromGPU", in SetProtoFromGPU()
178 LogMemory::PROTO_BUFFER_STEP_ID, in SetProtoFromGPU()
|
| D | gpu_event_mgr.h | 137 if (LogMemory::IsEnabled()) { in FreeMemory()
138 LogMemory::RecordRawDeallocation(iu.bufrec.operation, in FreeMemory()
|
| D | gpu_event_mgr.cc | 129 if (LogMemory::IsEnabled()) { in ~EventMgr()
130 LogMemory::RecordRawDeallocation(ue->bufrec.operation, in ~EventMgr()
|
| D | gpu_device.cc | 123 if (LogMemory::IsEnabled()) { in Reinitialize()
156 if (LogMemory::IsEnabled() && ret != nullptr) { in allocate()
157 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
163 if (LogMemory::IsEnabled() && buffer != nullptr) { in deallocate()
164 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_, in deallocate()
206 if (LogMemory::IsEnabled()) { in asyncFree()
207 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_, in asyncFree()
|
| D | gpu_process_state.cc | 247 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetGpuHostAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 112 if (LogMemory::IsEnabled() && !allocator->TracksAllocationSizes()) { in GetCPUAllocator()
|
| D | direct_session.cc | 784 if (LogMemory::IsEnabled()) { in Run()
785 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()
895 if (LogMemory::IsEnabled()) { in PRunSetup()
896 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()
1352 if (LogMemory::IsEnabled() || run_state_args->is_partial_run) { in GetOrCreateExecutors()
1834 if (LogMemory::IsEnabled()) { in RunCallable()
1835 LogMemory::RecordStep(step_id, run_state_args.handle); in RunCallable()
|
| D | graph_runner.cc | 177 args.step_id = LogMemory::CONSTANT_FOLDING_STEP_ID; in Run()
|
| D | executor.cc | 1380 log_memory_(LogMemory::IsEnabled()), in ExecutorState()
2051 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()
2062 LogMemory::RecordTensorOutput(ctx->op_kernel().name(), in ProcessOutputs()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 497 if (LogMemory::IsEnabled()) { in StartParallelExecutors()
498 LogMemory::RecordStep(args.step_id, handle); in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 158 bool LogMemory() const { return log_memory_; } in LogMemory() function
|
| D | context.cc | 68 log_memory_(LogMemory::IsEnabled()), in EagerContext()
|
| D | execute.cc | 452 ctx->GetRendezvous(), ctx->LogMemory(), flr, runner, in EagerLocalExecute()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 175 if (tensorflow::LogMemory::IsEnabled() && data != nullptr) { in allocate_tensor()
176 tensorflow::LogMemory::RecordRawAllocation( in allocate_tensor()
177 operation, tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, in allocate_tensor()
184 if (tensorflow::LogMemory::IsEnabled() && data != nullptr) { in deallocate_buffer()
185 tensorflow::LogMemory::RecordRawDeallocation( in deallocate_buffer()
187 tensorflow::LogMemory::EXTERNAL_TENSOR_ALLOCATION_STEP_ID, data, in deallocate_buffer()
|