Home
last modified time | relevance | path

Searched refs:AllocatorRetry (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dallocator_retry.h26 class AllocatorRetry {
28 AllocatorRetry();
53 inline void AllocatorRetry::NotifyDealloc() { in NotifyDealloc()
Dallocator_retry.cc49 AllocatorRetry::AllocatorRetry() : env_(Env::Default()) {} in AllocatorRetry() function in tensorflow::AllocatorRetry
51 void* AllocatorRetry::AllocateRaw( in AllocateRaw()
Dbfc_allocator.h520 AllocatorRetry retry_helper_;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_allocator_retry_test.cc58 AllocatorRetry retry_;