Public API for tf.config.experimental namespace.
class ClusterDeviceFilters
: Represent a collection of device filters for the remote workers in cluster.
class VirtualDeviceConfiguration
: Configuration class for a logical devices.
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.
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_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.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/config/experimental