Searched refs:GPUBFCAllocator (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator.cc | 22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()
55 bool GPUBFCAllocator::GetGarbageCollectionValue() { in GetGarbageCollectionValue()
76 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator() function in tensorflow::GPUBFCAllocator
78 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
80 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator() function in tensorflow::GPUBFCAllocator
85 GPUBFCAllocator::GetAllowGrowthValue(gpu_options), name, in GPUBFCAllocator()
86 GPUBFCAllocator::GetGarbageCollectionValue()) {} in GPUBFCAllocator()
|
| D | gpu_bfc_allocator.h | 35 class GPUBFCAllocator : public BFCAllocator {
37 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
39 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
41 ~GPUBFCAllocator() override {} in ~GPUBFCAllocator()
43 TF_DISALLOW_COPY_AND_ASSIGN(GPUBFCAllocator);
|
| D | gpu_bfc_allocator_test.cc | 106 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
135 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
193 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
228 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
234 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
239 GPUBFCAllocator a(GetParam()(1ull << 32), 1 << 30, "GPU_0_bfc"); in TEST_P()
248 GPUBFCAllocator a(GetParam()(1ull << 32), 2 << 20, "GPU_0_bfc"); in TEST_P()
263 GPUBFCAllocator a(GetParam()(1ull << 32), 1LL << 31, "GPU_0_bfc"); in TEST_P()
325 GPUBFCAllocator a(GetParam()(1ul << 62), 1UL << 60, "GPU_0_bfc"); in TEST_P()
326 GPUBFCAllocator b(GetParam()(1ul << 62), 1UL << 60, "GPU_0_bfc"); in TEST_P()
[all …]
|
| D | gpu_debug_allocator_test.cc | 52 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
80 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
116 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
148 GPUNanResetAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
192 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
235 GPUDebugAllocator a(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
246 new GPUDebugAllocator(new GPUBFCAllocator(sub_allocator, 1 << 30, ""), in TEST()
|
| D | gpu_process_state.h | 36 class GPUBFCAllocator; variable
147 GPUBFCAllocator* bfc_allocator;
|
| D | gpu_process_state.cc | 169 GPUBFCAllocator* gpu_bfc_allocator = in GetGPUAllocator()
170 new GPUBFCAllocator(sub_allocator, total_bytes, options, in GetGPUAllocator()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 174 // If true then extra work is done by GPUDevice and GPUBFCAllocator to
|