Searched refs:GpuAllocatorConfig (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 186 py::class_<GpuAllocatorConfig> alloc_config(m, "GpuAllocatorConfig"); in PYBIND11_MODULE()
188 .def_readwrite("kind", &GpuAllocatorConfig::kind) in PYBIND11_MODULE()
189 .def_readwrite("memory_fraction", &GpuAllocatorConfig::memory_fraction) in PYBIND11_MODULE()
190 .def_readwrite("preallocate", &GpuAllocatorConfig::preallocate); in PYBIND11_MODULE()
191 py::enum_<GpuAllocatorConfig::Kind>(alloc_config, "Kind") in PYBIND11_MODULE()
192 .value("DEFAULT", GpuAllocatorConfig::Kind::kDefault) in PYBIND11_MODULE()
193 .value("PLATFORM", GpuAllocatorConfig::Kind::kPlatform) in PYBIND11_MODULE()
194 .value("BFC", GpuAllocatorConfig::Kind::kBFC); in PYBIND11_MODULE()
245 [](bool asynchronous, const GpuAllocatorConfig& allocator_config, in PYBIND11_MODULE()
255 py::arg("allocator_config") = GpuAllocatorConfig(), in PYBIND11_MODULE()
|
| D | xla_client.py | 78 config = _xla.GpuAllocatorConfig()
80 config.kind = _xla.GpuAllocatorConfig.Kind.DEFAULT
82 config.kind = _xla.GpuAllocatorConfig.Kind.PLATFORM
84 config.kind = _xla.GpuAllocatorConfig.Kind.BFC
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.h | 34 struct GpuAllocatorConfig { struct
58 bool asynchronous, const GpuAllocatorConfig& allocator_config, argument
|
| D | gpu_device.cc | 147 const GpuAllocatorConfig& allocator_config, in GetGpuDeviceAllocator()
150 if (allocator_config.kind != GpuAllocatorConfig::Kind::kPlatform) { in GetGpuDeviceAllocator()
314 bool asynchronous, const GpuAllocatorConfig& allocator_config, in GetGpuClient()
|
| D | gpu_multistream_test.cc | 32 GetGpuClient(/*asynchronous=*/true, GpuAllocatorConfig(), in TEST()
|