Searched refs:DeviceHostAllocator (Results 1 – 4 of 4) sorted by relevance
26 class DeviceHostAllocator : public SubAllocator {29 explicit DeviceHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in DeviceHostAllocator() function37 ~DeviceHostAllocator() override {} in ~DeviceHostAllocator()68 TF_DISALLOW_COPY_AND_ASSIGN(DeviceHostAllocator);
33 new DeviceHostAllocator( in TEST()53 new DeviceHostAllocator( in TEST()88 new DeviceHostAllocator( in TEST()149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST()255 new DeviceHostAllocator( in TEST()
299 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator()
163 tensorflow::SubAllocator* sub_allocator = new tensorflow::DeviceHostAllocator( in GetGpuHostAllocator()