Searched refs:GpuHostAllocator (Results 1 – 3 of 3) sorted by relevance
26 class GpuHostAllocator : public SubAllocator {29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator() function37 ~GpuHostAllocator() override {} in ~GpuHostAllocator()64 TF_DISALLOW_COPY_AND_ASSIGN(GpuHostAllocator);
32 new GpuHostAllocator( in TEST()51 new GpuHostAllocator( in TEST()85 new GpuHostAllocator( in TEST()145 GpuHostAllocator* sub_allocator = new GpuHostAllocator( in TEST()250 new GpuHostAllocator( in TEST()
232 new GpuHostAllocator(se, numa_node, gpu_host_alloc_visitors_[numa_node], in GetGpuHostAllocator()