Searched refs:CPUSubAllocator (Results 1 – 1 of 1) sorted by relevance
186 return new CPUSubAllocator(new CPUAllocator); in CreateSubAllocator()190 class CPUSubAllocator : public SubAllocator { class in tensorflow::__anonab749f280111::CPUAllocatorFactory192 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator() function in tensorflow::__anonab749f280111::CPUAllocatorFactory::CPUSubAllocator