Searched defs:allocator_config (Results 1 – 3 of 3) sorted by relevance
268 se::Platform* platform, const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator()423 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient()
314 -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
1435 def allocator_option_enabled_fn(allocator_config, _, option): argument