| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator_test.cc | 54 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
57 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
87 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
90 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
149 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
152 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
188 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
191 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
198 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
201 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
[all …]
|
| D | gpu_debug_allocator_test.cc | 38 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
41 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
66 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
69 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
102 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
105 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
135 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
138 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
180 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST() local
184 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
[all …]
|
| D | gpu_bfc_allocator.cc | 58 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator() argument
60 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
62 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator() argument
66 : BFCAllocator(sub_allocator, total_memory, in GPUBFCAllocator()
|
| D | gpu_process_state.cc | 108 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in GetGPUAllocator() local
115 new GPUBFCAllocator(sub_allocator, total_bytes, options, in GetGPUAllocator()
149 sub_allocator, in GetGPUAllocator()
231 SubAllocator* sub_allocator = in GetGpuHostAllocator() local
244 new BFCAllocator(sub_allocator, gpu_host_mem_limit, in GetGpuHostAllocator()
254 sub_allocator, in GetGpuHostAllocator()
|
| D | gpu_bfc_allocator.h | 89 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
91 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
|
| D | pool_allocator_test.cc | 145 GpuHostAllocator* sub_allocator = new GpuHostAllocator( in TEST() local
150 sub_allocator, new NoopRounder, "pool"); in TEST()
|
| D | gpu_process_state.h | 140 SubAllocator* sub_allocator; // owned by allocator member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 78 SubAllocator* sub_allocator = in GetCPUAllocator() local
94 DCHECK(sub_allocator); in GetCPUAllocator()
96 new BFCAllocator(sub_allocator, cpu_mem_limit, true /*allow_growth*/, in GetCPUAllocator()
100 } else if (sub_allocator) { in GetCPUAllocator()
101 DCHECK(sub_allocator); in GetCPUAllocator()
104 sub_allocator, new NoopRounder, "cpu_pool"); in GetCPUAllocator()
109 DCHECK(!sub_allocator); in GetCPUAllocator()
118 if (!sub_allocator) { in GetCPUAllocator()
|
| 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 | 33 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator() argument
35 : sub_allocator_(sub_allocator), in BFCAllocator()
|
| D | bfc_allocator.h | 49 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
|