Home
last modified time | relevance | path

Searched refs:BasicCPUAllocator (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpool_allocator.h149 class BasicCPUAllocator : public SubAllocator {
151 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator() function
155 ~BasicCPUAllocator() override {} in ~BasicCPUAllocator()
164 TF_DISALLOW_COPY_AND_ASSIGN(BasicCPUAllocator);
Dpool_allocator.cc259 void* BasicCPUAllocator::Alloc(size_t alignment, size_t num_bytes) { in Alloc()
273 void BasicCPUAllocator::Free(void* ptr, size_t num_bytes) { in Free()
Dmkl_cpu_allocator.h44 class MklSubAllocator : public BasicCPUAllocator {
46 MklSubAllocator() : BasicCPUAllocator(port::kNUMANoAffinity, {}, {}) {} in MklSubAllocator()
Dprocess_state.cc80 ? new BasicCPUAllocator( in GetCPUAllocator()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dpool_allocator_test.cc103 new BasicCPUAllocator(0 /*numa_node*/, {}, {}), in TEST()