Searched defs:allocator_config (Results 1 – 2 of 2) sorted by relevance
147 const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator()314 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient()
247 int node_id) -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()