Searched refs:RecordRawAllocation (Results 1 – 6 of 6) sorted by relevance
95 static void RecordRawAllocation(const string& operation, int64 step_id,
77 void LogMemory::RecordRawAllocation(const string& operation, in RecordRawAllocation() function in tensorflow::LogMemory
45 tensorflow::LogMemory::RecordRawAllocation( in LogAllocation()
156 LogMemory::RecordRawAllocation("SetProtoFromGPU", in SetProtoFromGPU()
157 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
176 tensorflow::LogMemory::RecordRawAllocation( in allocate_tensor()