Searched refs:visible_device_list (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 36 const string& visible_device_list) { in ParseVisibleDeviceList() argument
38 if (visible_device_list.empty()) { in ParseVisibleDeviceList()
42 absl::StrSplit(visible_device_list, ','); in ParseVisibleDeviceList()
49 "'. visible_device_list = ", visible_device_list); in ParseVisibleDeviceList()
83 session_options.config.gpu_options().visible_device_list(); in CreateDevices()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 768 Status ParseVisibleDeviceList(const string& visible_device_list, in ParseVisibleDeviceList() argument
775 if (visible_device_list.empty()) { in ParseVisibleDeviceList()
783 str_util::Split(visible_device_list, ','); in ParseVisibleDeviceList()
790 "'. visible_device_list = ", visible_device_list); in ParseVisibleDeviceList()
809 visible_device_list); in ParseVisibleDeviceList()
831 if (!gpu_options.visible_device_list().empty() && in VerifyVirtualDeviceSettings()
1004 TF_RETURN_IF_ERROR(ParseVisibleDeviceList(gpu_options.visible_device_list(), in CreateDevices()
1112 CHECK(gpu_options.visible_device_list().empty() || in CreateDevices()
|
| D | gpu_device_test.cc | 69 const string& visible_device_list = "", in MakeSessionOptions() argument
76 gpu_options->set_visible_device_list(visible_device_list);
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops_test.py | 123 no_gpus = tf.GPUOptions(visible_device_list="")
128 one_gpu = tf.GPUOptions(visible_device_list=str(local_rank))
|
| D | mpi_allgather_test.py | 100 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
|
| D | mpi_allreduce_test.py | 105 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-g-p-u-options.pbtxt | 30 name: "visible_device_list"
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 76 string visible_device_list = 5; field
109 // "visible_device_list" above for more information.
115 // in "visible_device_list" above. Otherwise, the number of elements in the
117 // "visible_device_list" filtering if it is set), and the string represented
122 // visible_device_list = "1,0"
|
| /external/tensorflow/tensorflow/python/client/ |
| D | virtual_gpu_test.py | 91 visible_device_list=','.join(str(d) for d in self._visible_device_list),
|