Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.cc114 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anond7b9ad460111::MklCPUAllocatorFactory
/external/tensorflow/tensorflow/core/framework/
Dallocator_registry.h44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0;
Dallocator_registry.cc121 best_entry->factory->CreateSubAllocator(numa_node)); in GetSubAllocator()
Dallocator.cc185 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anonab749f280111::CPUAllocatorFactory