Searched refs:allocator_config (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 147 const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator() argument
150 if (allocator_config.kind != GpuAllocatorConfig::Kind::kPlatform) { in GetGpuDeviceAllocator()
153 allocator_config.memory_fraction, in GetGpuDeviceAllocator()
154 allocator_config.preallocate)); in GetGpuDeviceAllocator()
314 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient() argument
322 GetGpuDeviceAllocator(allocator_config, local_device_states)); in GetGpuClient()
|
| D | gpu_device.h | 58 bool asynchronous, const GpuAllocatorConfig& allocator_config,
|
| /external/llvm-project/llvm/utils/gn/secondary/compiler-rt/lib/scudo/standalone/ |
| D | BUILD.gn | 13 "allocator_config.h",
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 245 [](bool asynchronous, const GpuAllocatorConfig& allocator_config, in PYBIND11_MODULE()
250 GetGpuClient(asynchronous, allocator_config, in PYBIND11_MODULE()
|
| D | xla_client.py | 91 allocator_config=config,
|
| /external/llvm-project/compiler-rt/lib/scudo/standalone/ |
| D | CMakeLists.txt | 53 allocator_config.h
|