Searched refs:memory_fraction (Results 1 – 4 of 4) sorted by relevance
72 memory_fraction = os.getenv('XLA_PYTHON_CLIENT_MEM_FRACTION')85 if memory_fraction:86 config.memory_fraction = float(memory_fraction)
189 .def_readwrite("memory_fraction", &GpuAllocatorConfig::memory_fraction) in PYBIND11_MODULE()
45 double memory_fraction = 0.9; member
92 double memory_fraction, bool preallocate) { in CreateBFCAllocator() argument123 enable_unified_memory ? total_memory : free_memory * memory_fraction; in CreateBFCAllocator()153 allocator_config.memory_fraction, in GetGpuDeviceAllocator()