Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.cc111 Allocator* CreateAllocator() override { return new MklCPUAllocator; } in CreateAllocator() function in tensorflow::__anond7b9ad460111::MklCPUAllocatorFactory
/external/tensorflow/tensorflow/core/framework/
Dallocator_registry.h39 virtual Allocator* CreateAllocator() = 0;
Dallocator_registry.cc81 best_entry->allocator.reset(best_entry->factory->CreateAllocator()); in GetAllocator()
Dallocator.cc183 Allocator* CreateAllocator() override { return new CPUAllocator; } in CreateAllocator() function in tensorflow::__anonab749f280111::CPUAllocatorFactory