| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 50 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
52 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
77 DeviceMemAllocator* sub_allocator = in TEST()
80 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
113 DeviceMemAllocator* sub_allocator = in TEST()
116 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
146 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
148 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
189 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
192 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
[all …]
|
| D | gpu_bfc_allocator.cc | 76 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator() argument
78 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
80 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator() argument
84 : BFCAllocator(sub_allocator, total_memory, in GPUBFCAllocator()
|
| D | gpu_bfc_allocator.h | 37 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
39 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
|
| D | gpu_process_state.cc | 166 auto* sub_allocator = in GetGPUAllocator() local
170 new GPUBFCAllocator(sub_allocator, total_bytes, options, in GetGPUAllocator()
206 gpu_bfc_allocator, sub_allocator, in GetGPUAllocator()
299 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator() local
313 new BFCAllocator(sub_allocator, gpu_host_mem_limit, in GetGpuHostAllocator()
323 nullptr, sub_allocator, in GetGpuHostAllocator()
|
| D | pool_allocator_test.cc | 149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST() local
154 sub_allocator, new NoopRounder, "pool"); in TEST()
|
| D | gpu_process_state.h | 148 SubAllocator* sub_allocator; // owned by allocator member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 86 SubAllocator* sub_allocator = in GetCPUAllocator() local
102 DCHECK(sub_allocator); in GetCPUAllocator()
104 new BFCAllocator(sub_allocator, cpu_mem_limit, /*allow_growth=*/true, in GetCPUAllocator()
108 } else if (sub_allocator) { in GetCPUAllocator()
109 DCHECK(sub_allocator); in GetCPUAllocator()
112 sub_allocator, new NoopRounder, "cpu_pool"); in GetCPUAllocator()
117 DCHECK(!sub_allocator); in GetCPUAllocator()
130 if (!sub_allocator) { in GetCPUAllocator()
|
| D | bfc_allocator_test.cc | 53 FakeSubAllocator* sub_allocator = new FakeSubAllocator; in BM_Allocator() local
54 BFCAllocator bfc_allocator(sub_allocator, 1 << 30, false, "GPU_0_bfc"); in BM_Allocator()
|
| D | mkl_cpu_allocator.h | 55 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator() argument
57 : sub_allocator_(sub_allocator), name_(name) { in MklSmallSizeAllocator()
|
| D | bfc_allocator.cc | 40 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator() argument
44 coalesce_regions_(sub_allocator->SupportsCoalescing()), in BFCAllocator()
45 sub_allocator_(sub_allocator), in BFCAllocator()
|
| D | bfc_allocator.h | 52 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 108 auto sub_allocator = absl::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator() local
134 sub_allocator.release(), allocator_memory, in CreateBFCAllocator()
163 tensorflow::SubAllocator* sub_allocator = new tensorflow::DeviceHostAllocator( in GetGpuHostAllocator() local
168 sub_allocator, kGpuHostMemoryLimitBytes, /*allow_growth=*/true, in GetGpuHostAllocator()
|