Home
last modified time | relevance | path

Searched refs:visible_device_list (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_utils.h67 const string& visible_device_list, const int visible_device_count, in ParseVisibleDeviceList() argument
73 if (visible_device_list.empty()) { in ParseVisibleDeviceList()
79 str_util::Split(visible_device_list, ','); in ParseVisibleDeviceList()
87 "'. visible_device_list = ", visible_device_list); in ParseVisibleDeviceList()
106 visible_device_list); in ParseVisibleDeviceList()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc26 absl::string_view visible_device_list) { in ParseVisibleDeviceList() argument
28 if (visible_device_list.empty()) { in ParseVisibleDeviceList()
32 absl::StrSplit(visible_device_list, ','); in ParseVisibleDeviceList()
39 "'. visible_device_list = ", visible_device_list); in ParseVisibleDeviceList()
95 flr->config_proto()->gpu_options().visible_device_list(); in BuildXlaCompilationCache()
Dxla_platform_info.h88 absl::string_view visible_device_list);
Dxla_gpu_device.cc118 session_options.config.gpu_options().visible_device_list(); in CreateDevices()
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py790 gpu_options=config_pb2.GPUOptions(visible_device_list='0'))
792 self.assertEqual(new_config.gpu_options.visible_device_list, '0')
811 gpu_options=config_pb2.GPUOptions(visible_device_list=''))
815 self.assertEqual(new_config.gpu_options.visible_device_list,
821 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count)))
830 visible_device_list=str(gpu_count - 1)))
833 self.assertEqual(new_config.gpu_options.visible_device_list,
/external/tensorflow/tensorflow/python/eager/
Dcontext.py1182 visible_device_list = []
1200 visible_device_list.append(str(gpu_index))
1234 visible_device_list=",".join(visible_device_list),
1529 visible_indices = self._config.gpu_options.visible_device_list
1575 visible_device_list = []
1577 visible_device_list = [
1581 visible_device_list += devices
1583 if self._visible_device_list == visible_device_list:
1590 self._visible_device_list = visible_device_list
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-059.md11 flr->config_proto()->gpu_options().visible_device_list();
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-g-p-u-options.pbtxt30 name: "visible_device_list"
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc72 const string& visible_device_list = "", in MakeSessionOptions() argument
82 gpu_options->set_visible_device_list(visible_device_list);
Dgpu_device.cc949 if (!gpu_options.visible_device_list().empty() && in VerifyVirtualDeviceSettings()
1295 gpu_options.visible_device_list(), gpu_manager->VisibleDeviceCount(), in CreateDevices()
1450 CHECK(gpu_options.visible_device_list().empty() || in CreateDevices()
/external/tensorflow/tensorflow/python/client/
Dvirtual_gpu_test.py87 visible_device_list=','.join(str(d) for d in self._visible_device_list),
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_factory.cc181 device_options.visible_device_list(), platform->VisibleDeviceCount(), in CreateDevices()
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto79 string visible_device_list = 5; field
112 // "visible_device_list" above for more information.
136 // in "visible_device_list" above. Otherwise, the number of elements in the
138 // "visible_device_list" filtering if it is set), and the string represented
145 // visible_device_list = "1,0"
155 // visible_device_list = "1,0"