Public API for tf.config.experimental namespace.
Classes
class ClusterDeviceFilters: Represent a collection of device filters for the remote workers in cluster.
class VirtualDeviceConfiguration: Configuration class for a logical devices.
Functions
disable_mlir_bridge(...): Disables experimental MLIR-Based TensorFlow Compiler Bridge.
disable_mlir_graph_optimization(...): Disables experimental MLIR-Based TensorFlow Compiler Optimizations.
enable_mlir_bridge(...): Enables experimental MLIR-Based TensorFlow Compiler Bridge.
enable_mlir_graph_optimization(...): Enables experimental MLIR-Based TensorFlow Compiler Optimizations.
enable_tensor_float_32_execution(...): Enable or disable the use of TensorFloat-32 on supported hardware.
get_device_details(...): Returns details about a physical devices.
get_device_policy(...): Gets the current device policy.
get_memory_growth(...): Get if memory growth is enabled for a PhysicalDevice.
get_memory_info(...): Get memory info for the chosen device, as a dict.
get_memory_usage(...): Get the current memory usage, in bytes, for the chosen device. (deprecated)
get_synchronous_execution(...): Gets whether operations are executed synchronously or asynchronously.
get_virtual_device_configuration(...): Get the virtual device configuration for a tf.config.PhysicalDevice.
get_visible_devices(...): Get the list of visible physical devices.
list_logical_devices(...): Return a list of logical devices created by runtime.
list_physical_devices(...): Return a list of physical devices visible to the host runtime.
reset_memory_stats(...): Resets the tracked memory stats for the chosen device.
set_device_policy(...): Sets the current thread device policy.
set_memory_growth(...): Set if memory growth should be enabled for a PhysicalDevice.
set_synchronous_execution(...): Specifies whether operations are executed synchronously or asynchronously.
set_virtual_device_configuration(...): Set the logical device configuration for a tf.config.PhysicalDevice.
set_visible_devices(...): Set the list of visible devices.
tensor_float_32_execution_enabled(...): Returns whether TensorFloat-32 is enabled.