Searched refs:DeviceMemAllocator (Results 1 – 5 of 5) sorted by relevance
50 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST()77 DeviceMemAllocator* sub_allocator = in TEST()78 new DeviceMemAllocator(stream_exec, platform_gpu_id, in TEST()113 DeviceMemAllocator* sub_allocator = in TEST()114 new DeviceMemAllocator(stream_exec, platform_gpu_id, in TEST()146 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST()189 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST()232 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST()242 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST()
98 return new DeviceMemAllocator(executor, platform_gpu_id, in CreateSubAllocator()128 return new DeviceMemAllocator( in CreateSubAllocator()
80 return new DeviceMemAllocator( in CreateGPUMemAllocator()
26 class DeviceMemAllocator : public SubAllocator {31 explicit DeviceMemAllocator(se::StreamExecutor* stream_exec, in DeviceMemAllocator() function42 ~DeviceMemAllocator() override {} in ~DeviceMemAllocator()78 TF_DISALLOW_COPY_AND_ASSIGN(DeviceMemAllocator);
108 auto sub_allocator = absl::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator()