Searched refs:per_process_gpu_memory_fraction (Results 1 – 16 of 16) sorted by relevance
181 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))186 per_process_gpu_memory_fraction)200 per_process_gpu_memory_fraction)
111 if (options.per_process_gpu_memory_fraction() > 1.0 || in CreateSubAllocator()145 (options.per_process_gpu_memory_fraction() > 1.0 || in CreateSubAllocator()
938 if (gpu_options.per_process_gpu_memory_fraction() > 0) { in VerifyVirtualDeviceSettings()1099 const double per_process_gpu_memory_fraction = in SingleVirtualDeviceMemoryLimit() local1100 gpu_options.per_process_gpu_memory_fraction(); in SingleVirtualDeviceMemoryLimit()1103 if ((per_process_gpu_memory_fraction > 1.0 || in SingleVirtualDeviceMemoryLimit()1111 if (per_process_gpu_memory_fraction == 0) { in SingleVirtualDeviceMemoryLimit()1119 allocated_memory = total_memory * per_process_gpu_memory_fraction; in SingleVirtualDeviceMemoryLimit()
73 double per_process_gpu_memory_fraction = 0, int gpu_device_count = 1, in MakeSessionOptions() argument84 per_process_gpu_memory_fraction);
106 per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction))
449 per_process_gpu_memory_fraction=gpu_memory_fraction))
6 name: "per_process_gpu_memory_fraction"
74 config.gpu_options.per_process_gpu_memory_fraction = 0.3
111 bool use_unified_memory = options.per_process_gpu_memory_fraction() > 1.0 || in GetPluggableDeviceAllocator()
96 device_options.per_process_gpu_memory_fraction(); in SingleVirtualDeviceMemoryLimit()
25 config.gpu_options.per_process_gpu_memory_fraction = 0.44
37 double per_process_gpu_memory_fraction = 1; field167 // 1. It's invalid to set both this and "per_process_gpu_memory_fraction"175 // per_process_gpu_memory_fraction option is greater than 1.0, then unified177 // per_process_gpu_memory_fraction field for more details and requirements
132 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
217 config.gpu_options.per_process_gpu_memory_fraction = 0.3
71 config.gpu_options.per_process_gpu_memory_fraction = 1e-2
859 double per_process_gpu_memory_fraction() const; in Swap()4640 inline double GPUOptions::per_process_gpu_memory_fraction() const { in per_process_gpu_memory_fraction() function