Searched refs:RecordRawAllocation (Results 1 – 6 of 6) sorted by relevance
95 static void RecordRawAllocation(const string& operation, int64 step_id,
76 void LogMemory::RecordRawAllocation(const string& operation, in RecordRawAllocation() function in tensorflow::LogMemory
46 tensorflow::LogMemory::RecordRawAllocation( in LogAllocation()
155 LogMemory::RecordRawAllocation("SetProtoFromGPU", in SetProtoFromGPU()
159 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
40 LogMemory::RecordRawAllocation( in allocate_tensor()