MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ev_allocator

Function ev_allocator

tensorflow/core/framework/allocator.cc:119–128  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

117}
118
119Allocator* ev_allocator() {
120 static Allocator* ev_alloc = DisableEVAllocatorFromEnvironment() ?
121 cpu_allocator() : AllocatorFactoryRegistry::singleton()->GetEVAllocator();
122
123 if (ev_alloc && cpu_allocator_collect_full_stats &&
124 !ev_alloc->TracksAllocationSizes()) {
125 ev_alloc = new TrackingAllocator(ev_alloc, true);
126 }
127 return ev_alloc;
128}
129
130Allocator* gpu_ev_allocator() {
131 static Allocator* ev_alloc =

Callers 15

TESTFunction · 0.85
GetSnapshotMethod · 0.85
GetShardedSnapshotMethod · 0.85
RestoreSSDMethod · 0.85
SaveMethod · 0.85
InitMethod · 0.85
TESTFunction · 0.85
TestCompactionFunction · 0.85

Calls 4

cpu_allocatorFunction · 0.85
GetEVAllocatorMethod · 0.80
TracksAllocationSizesMethod · 0.45

Tested by 5

TESTFunction · 0.68
TESTFunction · 0.68
TestCompactionFunction · 0.68
TestReadEmbFileFunction · 0.68
CreateEmbeddingVarFunction · 0.68