Searched refs:CPUAllocator (Results 1 – 1 of 1) sorted by relevance
102 class CPUAllocator : public Allocator { class104 CPUAllocator() in CPUAllocator() function in tensorflow::__anonab749f280111::CPUAllocator108 ~CPUAllocator() override {} in ~CPUAllocator()178 TF_DISALLOW_COPY_AND_ASSIGN(CPUAllocator);183 Allocator* CreateAllocator() override { return new CPUAllocator; } in CreateAllocator()186 return new CPUSubAllocator(new CPUAllocator); in CreateSubAllocator()192 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()204 CPUAllocator* cpu_allocator_;