Searched refs:per_process_gpu_memory_fraction (Results 1 – 18 of 18) sorted by relevance
6 name: "per_process_gpu_memory_fraction"
60 gpu_options = config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.50)
109 per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction))
88 per_process_gpu_memory_fraction=cls.PER_PROC_GPU_MEMORY_FRACTION))
453 per_process_gpu_memory_fraction=gpu_memory_fraction))
70 double per_process_gpu_memory_fraction = 0, int gpu_device_count = 1, in MakeSessionOptions() argument78 per_process_gpu_memory_fraction);
820 if (gpu_options.per_process_gpu_memory_fraction() > 0) { in VerifyVirtualDeviceSettings()900 const double per_process_gpu_memory_fraction = in SingleVirtualDeviceMemoryLimit() local901 gpu_options.per_process_gpu_memory_fraction(); in SingleVirtualDeviceMemoryLimit()902 if (per_process_gpu_memory_fraction > 1.0 || in SingleVirtualDeviceMemoryLimit()916 if (per_process_gpu_memory_fraction == 0) { in SingleVirtualDeviceMemoryLimit()923 allocated_memory = total_memory * per_process_gpu_memory_fraction; in SingleVirtualDeviceMemoryLimit()
111 (options.per_process_gpu_memory_fraction() > 1.0 || in GetGPUAllocator()
186 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))191 per_process_gpu_memory_fraction)205 per_process_gpu_memory_fraction)
309 per_process_gpu_memory_fraction=gpu_memory_fraction)
735 return self._config.gpu_options.per_process_gpu_memory_fraction743 self._config.gpu_options.per_process_gpu_memory_fraction = fraction
34 double per_process_gpu_memory_fraction = 1; field131 // 1. It's invalid to set both this and "per_process_gpu_memory_fraction"139 // per_process_gpu_memory_fraction option is greater than 1.0, then unified141 // per_process_gpu_memory_fraction field for more details and requirements
141 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
167 config.gpu_options.per_process_gpu_memory_fraction = 0.3
300 config.gpu_options.per_process_gpu_memory_fraction = 0.3
71 config.gpu_options.per_process_gpu_memory_fraction = 1e-2
314 expected_config_proto.gpu_options.per_process_gpu_memory_fraction = 0.314
2578 config.gpu_options.per_process_gpu_memory_fraction = 0.3