HomeSort by relevance Sort by last modified time
    Searched refs:per_process_gpu_memory_fraction (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
plugin_test.py 60 gpu_options = config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.50)
  /external/tensorflow/tensorflow/python/debug/lib/
grpc_tensorflow_server.py 109 per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction))
dist_session_debug_grpc_test.py 88 per_process_gpu_memory_fraction=cls.PER_PROC_GPU_MEMORY_FRACTION))
grpc_debug_test_server.py 453 per_process_gpu_memory_fraction=gpu_memory_fraction))
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device_test.cc 70 double per_process_gpu_memory_fraction = 0, int gpu_device_count = 1,
78 per_process_gpu_memory_fraction);
126 status, "It's invalid to set per_process_gpu_memory_fraction");
gpu_device.cc 900 const double per_process_gpu_memory_fraction = local
    [all...]
gpu_process_state.cc 111 (options.per_process_gpu_memory_fraction() > 1.0 ||
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
run_config.py 309 per_process_gpu_memory_fraction=gpu_memory_fraction)
  /external/tensorflow/tensorflow/python/training/
server_lib_test.py 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)
  /external/tensorflow/tensorflow/contrib/distribute/python/
multi_worker_test_base.py 141 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
strategy_test_lib.py 167 config.gpu_options.per_process_gpu_memory_fraction = 0.3
  /external/tensorflow/tensorflow/python/eager/
context.py 735 return self._config.gpu_options.per_process_gpu_memory_fraction
743 self._config.gpu_options.per_process_gpu_memory_fraction = fraction
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
sparse_xent_op_test.py 300 config.gpu_options.per_process_gpu_memory_fraction = 0.3
  /external/tensorflow/tensorflow/python/kernel_tests/signal/
fft_ops_test.py 71 config.gpu_options.per_process_gpu_memory_fraction = 1e-2
  /external/tensorflow/tensorflow/contrib/learn/python/learn/
experiment_test.py 314 expected_config_proto.gpu_options.per_process_gpu_memory_fraction = 0.314
    [all...]
  /external/tensorflow/tensorflow/python/framework/
test_util.py     [all...]

Completed in 805 milliseconds