Searched refs:CpuAllocator (Results 1 – 1 of 1) sorted by relevance
176 class CpuAllocator : public tensorflow::Allocator { class178 CpuAllocator() = default;230 host_memory_allocator_ = std::make_unique<CpuAllocator>(); in PjRtStreamExecutorClient()