Searched refs:allocator_name (Results 1 – 13 of 13) sorted by relevance
37 string allocator_name = 2; field73 string allocator_name = 6; field88 string allocator_name = 4; field
59 const string& allocator_name) { in RecordTensorDeallocation() argument62 deallocation.set_allocator_name(allocator_name); in RecordTensorDeallocation()
80 const string& allocator_name);
17 string allocator_name = 3; field
21 string allocator_name = 1; field
245 const char* allocator_name) { in DumpHeapUsage() argument247 DCHECK_EQ(0ul, heap_dumps_.count(allocator_name)); in DumpHeapUsage()250 heap_dumps_[allocator_name] = std::move(heap_dump); in DumpHeapUsage()254 allocator_name); in DumpHeapUsage()
125 const char* allocator_name);
97 if (mem.allocator_name().find("GPU") == mem.allocator_name().npos) { in AddMemoryStats()
501 allocator_name = allocation.allocator_name503 allocator_name, num_bytes)
303 if (err.find(alloc.first->allocator_name()) == err.npos) { in ReportAllocsOnResourceExhausted()307 std::make_pair(dev_stat.first, alloc.first->allocator_name()); in ReportAllocsOnResourceExhausted()
89 strings::StrCat(memory, "[", all.allocator_name(), in SetTimelineLabel()93 memory = strings::StrCat(memory, "[", all.allocator_name(), in SetTimelineLabel()
204 const std::string& allocator_name);
1419 const std::string& allocator_name) { in InsertMemoryRegion() argument1425 << " SIZE: 0x" << length << " (" << allocator_name << ")."; in InsertMemoryRegion()