Home
last modified time | relevance | path

Searched refs:GPUBFCAllocator (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator.cc25 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()
58 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator() function in tensorflow::GPUBFCAllocator
60 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
62 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator() function in tensorflow::GPUBFCAllocator
67 GPUBFCAllocator::GetAllowGrowthValue(gpu_options), name) {} in GPUBFCAllocator()
Dgpu_bfc_allocator_test.cc57 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
90 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
152 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
191 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
201 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
210 GPUBFCAllocator a(sub_allocator, 1 << 30, "GPU_0_bfc"); in TEST()
223 GPUBFCAllocator a(sub_allocator, 1 << 20, "GPU_0_bfc"); in TEST()
242 GPUBFCAllocator a(sub_allocator, 1LL << 31, "GPU_0_bfc"); in TEST()
308 GPUBFCAllocator a(sub_allocator, 1UL << 60, "GPU_0_bfc"); in TEST()
312 GPUBFCAllocator b(sub_allocator, 1UL << 60, "GPU_0_bfc"); in TEST()
[all …]
Dgpu_bfc_allocator.h87 class GPUBFCAllocator : public BFCAllocator {
89 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
91 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
93 ~GPUBFCAllocator() override {} in ~GPUBFCAllocator()
95 TF_DISALLOW_COPY_AND_ASSIGN(GPUBFCAllocator);
Dgpu_debug_allocator_test.cc41 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
69 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
105 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
138 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
184 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
229 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
240 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
Dgpu_process_state.cc114 GPUBFCAllocator* gpu_bfc_allocator = in GetGPUAllocator()
115 new GPUBFCAllocator(sub_allocator, total_bytes, options, in GetGPUAllocator()
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto160 // If true then extra work is done by GPUDevice and GPUBFCAllocator to