Searched refs:BasicCPUAllocator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.h | 150 class BasicCPUAllocator : public SubAllocator {
152 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator() function
156 ~BasicCPUAllocator() override {} in ~BasicCPUAllocator()
168 TF_DISALLOW_COPY_AND_ASSIGN(BasicCPUAllocator);
|
| D | pool_allocator.cc | 260 void* BasicCPUAllocator::Alloc(size_t alignment, size_t num_bytes, in Alloc()
276 void BasicCPUAllocator::Free(void* ptr, size_t num_bytes) { in Free()
|
| D | mkl_cpu_allocator.h | 44 class MklSubAllocator : public BasicCPUAllocator {
46 MklSubAllocator() : BasicCPUAllocator(port::kNUMANoAffinity, {}, {}) {} in MklSubAllocator()
|
| D | process_state.cc | 88 ? new BasicCPUAllocator( in GetCPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 106 new BasicCPUAllocator(0 /*numa_node*/, {}, {}), in TEST()
|