Searched refs:AllocatorRetry (Results 1 – 4 of 4) sorted by relevance
26 class AllocatorRetry {28 AllocatorRetry();53 inline void AllocatorRetry::NotifyDealloc() { in NotifyDealloc()
49 AllocatorRetry::AllocatorRetry() : env_(Env::Default()) {} in AllocatorRetry() function in tensorflow::AllocatorRetry51 void* AllocatorRetry::AllocateRaw( in AllocateRaw()
520 AllocatorRetry retry_helper_;
58 AllocatorRetry retry_;