Module: tf.config.experimental

TensorFlow 1 version

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.

enable_mlir_bridge(...): Enables experimental MLIR-Based TensorFlow Compiler Bridge.

get_device_policy(...): Gets the current device policy.

get_memory_growth(...): Get if memory growth is enabled for a PhysicalDevice.

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.

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.