Module: tf.config.experimental

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(...)</