Searched refs:bfc_allocator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator_test.cc | 54 BFCAllocator bfc_allocator(sub_allocator, 1 << 30, false, "GPU_0_bfc"); in BM_Allocator() local
62 long_lived[i] = bfc_allocator.AllocateRaw(1, kAllocSize); in BM_Allocator()
75 short_lived[i] = bfc_allocator.AllocateRaw(1, kAllocSize); in BM_Allocator()
78 bfc_allocator.DeallocateRaw(short_lived[deallocation_order[i]]); in BM_Allocator()
|
| D | BUILD | 214 "bfc_allocator.h",
1029 ":bfc_allocator",
1195 ":bfc_allocator",
1571 ":bfc_allocator",
1683 name = "bfc_allocator",
1687 "bfc_allocator.cc",
1689 hdrs = ["bfc_allocator.h"],
2726 ":bfc_allocator",
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.h | 147 GPUBFCAllocator* bfc_allocator; member
|
| D | gpu_process_state.cc | 237 allocator_parts.bfc_allocator->SetTimingCounter(timing_counter); in GPUAllocatorCounter()
|
| D | BUILD | 197 "//tensorflow/core/common_runtime:bfc_allocator",
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | BUILD | 310 "//tensorflow/core/common_runtime:bfc_allocator",
|