Searched refs:per_process_gpu_memory_fraction (Results 1 – 15 of 15) sorted by relevance
109 per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction))
90 per_process_gpu_memory_fraction=cls.PER_PROC_GPU_MEMORY_FRACTION))
453 per_process_gpu_memory_fraction=gpu_memory_fraction))
6 name: "per_process_gpu_memory_fraction"
96 if (options.per_process_gpu_memory_fraction() > 1.0 || in CreateSubAllocator()130 (options.per_process_gpu_memory_fraction() > 1.0 || in CreateSubAllocator()
833 if (gpu_options.per_process_gpu_memory_fraction() > 0) { in VerifyVirtualDeviceSettings()969 const double per_process_gpu_memory_fraction = in SingleVirtualDeviceMemoryLimit() local970 gpu_options.per_process_gpu_memory_fraction(); in SingleVirtualDeviceMemoryLimit()976 if (per_process_gpu_memory_fraction > 1.0 || in SingleVirtualDeviceMemoryLimit()985 if (per_process_gpu_memory_fraction == 0) { in SingleVirtualDeviceMemoryLimit()992 allocated_memory = total_memory * per_process_gpu_memory_fraction; in SingleVirtualDeviceMemoryLimit()
72 double per_process_gpu_memory_fraction = 0, int gpu_device_count = 1, in MakeSessionOptions() argument81 per_process_gpu_memory_fraction);
185 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))190 per_process_gpu_memory_fraction)204 per_process_gpu_memory_fraction)
25 config.gpu_options.per_process_gpu_memory_fraction = 0.44
340 config.gpu_options.per_process_gpu_memory_fraction = 0.3
289 config.gpu_options.per_process_gpu_memory_fraction = 0.3
36 double per_process_gpu_memory_fraction = 1; field145 // 1. It's invalid to set both this and "per_process_gpu_memory_fraction"153 // per_process_gpu_memory_fraction option is greater than 1.0, then unified155 // per_process_gpu_memory_fraction field for more details and requirements
165 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
200 config.gpu_options.per_process_gpu_memory_fraction = 0.3
81 config.gpu_options.per_process_gpu_memory_fraction = 1e-2