Searched refs:CreateSubAllocator (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ | ||
D | threadpool_device.cc | 114 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anond7b9ad460111::MklCPUAllocatorFactory |
/external/tensorflow/tensorflow/core/framework/ | ||
D | allocator_registry.h | 44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0; |
D | allocator_registry.cc | 121 best_entry->factory->CreateSubAllocator(numa_node)); in GetSubAllocator() |
D | allocator.cc | 185 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anonab749f280111::CPUAllocatorFactory |