Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dallocator_registry.cc24 AllocatorFactoryRegistry* AllocatorFactoryRegistry::singleton() { in singleton()
25 static AllocatorFactoryRegistry* singleton = new AllocatorFactoryRegistry; in singleton()
29 const AllocatorFactoryRegistry::FactoryEntry*
30 AllocatorFactoryRegistry::FindEntry(const string& name, int priority) const { in FindEntry()
39 void AllocatorFactoryRegistry::Register(const char* source_file, in Register()
68 Allocator* AllocatorFactoryRegistry::GetAllocator() { in GetAllocator()
90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
Dallocator_registry.h61 class AllocatorFactoryRegistry {
63 AllocatorFactoryRegistry() {} in AllocatorFactoryRegistry() function
64 ~AllocatorFactoryRegistry() {} in ~AllocatorFactoryRegistry()
82 static AllocatorFactoryRegistry* singleton();
111 TF_DISALLOW_COPY_AND_ASSIGN(AllocatorFactoryRegistry);
118 AllocatorFactoryRegistry::singleton()->Register(file, line, name, priority, in AllocatorFactoryRegistration()
Dallocator.cc70 AllocatorFactoryRegistry::singleton()->GetAllocator(); in cpu_allocator_base()
87 AllocatorFactoryRegistry::singleton()->process_state(); in cpu_allocator()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_state.cc40 AllocatorFactoryRegistry::singleton()->process_state_ = instance; in singleton()