bef9713188
This API can get details about physical devices. Right now, only GPUs are supported, and the only fields are "name" and "compute_capability". The primary motivation is to determine whether mixed precision will run well, as it only results in significant speedups on GPUs with compute capability 7.0 and greater. In general, it's rare that querying device details is necessary, as TensorFlow runs most ops well on all devices, but mixed precision is an exception. PiperOrigin-RevId: 315943445 Change-Id: I077fdc8f87a713ace74037fd2d82eede48740067
80 lines
2.5 KiB
Plaintext
80 lines
2.5 KiB
Plaintext
path: "tensorflow.config.experimental"
|
|
tf_module {
|
|
member {
|
|
name: "ClusterDeviceFilters"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "VirtualDeviceConfiguration"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member_method {
|
|
name: "disable_mlir_bridge"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "disable_mlir_graph_optimization"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "enable_mlir_bridge"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "enable_mlir_graph_optimization"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_device_details"
|
|
argspec: "args=[\'device\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_device_policy"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_memory_growth"
|
|
argspec: "args=[\'device\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_synchronous_execution"
|
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_virtual_device_configuration"
|
|
argspec: "args=[\'device\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "get_visible_devices"
|
|
argspec: "args=[\'device_type\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "list_logical_devices"
|
|
argspec: "args=[\'device_type\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "list_physical_devices"
|
|
argspec: "args=[\'device_type\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
member_method {
|
|
name: "set_device_policy"
|
|
argspec: "args=[\'device_policy\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "set_memory_growth"
|
|
argspec: "args=[\'device\', \'enable\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "set_synchronous_execution"
|
|
argspec: "args=[\'enable\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "set_virtual_device_configuration"
|
|
argspec: "args=[\'device\', \'logical_devices\'], varargs=None, keywords=None, defaults=None"
|
|
}
|
|
member_method {
|
|
name: "set_visible_devices"
|
|
argspec: "args=[\'devices\', \'device_type\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
|
}
|
|
}
|