Home
last modified time | relevance | path

Searched refs:GPUMemAllocator (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc54 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
87 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
149 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
188 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
198 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
207 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
219 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
239 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
305 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
309 sub_allocator = new GPUMemAllocator( in TEST()
[all …]
Dgpu_debug_allocator_test.cc38 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
66 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
102 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
135 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
180 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
226 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
236 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
Dgpu_bfc_allocator.h35 class GPUMemAllocator : public SubAllocator {
40 explicit GPUMemAllocator(se::StreamExecutor* stream_exec, in GPUMemAllocator() function
50 ~GPUMemAllocator() override {} in ~GPUMemAllocator()
82 TF_DISALLOW_COPY_AND_ASSIGN(GPUMemAllocator);
89 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
91 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
Dgpu_bfc_allocator.cc58 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator()
62 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator()
Dgpu_process_state.cc108 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in GetGPUAllocator()