Searched refs:CreateAllocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | threadpool_device.cc | 111 Allocator* CreateAllocator() override { return new MklCPUAllocator; } in CreateAllocator() function in tensorflow::__anond7b9ad460111::MklCPUAllocatorFactory |
| /external/tensorflow/tensorflow/core/framework/ | ||
| D | allocator_registry.h | 39 virtual Allocator* CreateAllocator() = 0; |
| D | allocator_registry.cc | 81 best_entry->allocator.reset(best_entry->factory->CreateAllocator()); in GetAllocator() |
| D | allocator.cc | 183 Allocator* CreateAllocator() override { return new CPUAllocator; } in CreateAllocator() function in tensorflow::__anonab749f280111::CPUAllocatorFactory |