Searched defs:allocator_config (Results 1 – 2 of 2) sorted by relevance
290 se::Platform* platform, const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator()484 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient()
284 int node_id) -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()