Home
last modified time | relevance | path

Searched defs:allocator_config (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dgpu_device.cc268 se::Platform* platform, const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator()
423 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla.cc314 -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1435 def allocator_option_enabled_fn(allocator_config, _, option): argument