Searched refs:CPUSubAllocator (Results 1 – 1 of 1) sorted by relevance
150 return new CPUSubAllocator(new CPUAllocator); in CreateSubAllocator()154 class CPUSubAllocator : public SubAllocator { class in tensorflow::__anon435a67570111::CPUAllocatorFactory156 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator() function in tensorflow::__anon435a67570111::CPUAllocatorFactory::CPUSubAllocator