Home
last modified time | relevance | path

Searched refs:RecordRawAllocation (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.h95 static void RecordRawAllocation(const string& operation, int64 step_id,
Dlog_memory.cc76 void LogMemory::RecordRawAllocation(const string& operation, in RecordRawAllocation() function in tensorflow::LogMemory
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map.cc46 tensorflow::LogMemory::RecordRawAllocation( in LogAllocation()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc155 LogMemory::RecordRawAllocation("SetProtoFromGPU", in SetProtoFromGPU()
Dgpu_device.cc159 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc40 LogMemory::RecordRawAllocation( in allocate_tensor()