Module: tf.keras.dtensor.experimental

Public API for tf.keras.dtensor.experimental namespace.

Modules

optimizers module: Public API for tf.keras.dtensor.experimental.optimizers namespace.

Classes

class LayoutMap: A dict-like object that maps string to Layout instances.

Functions

layout_map_scope(...): Apply the layout to all the tf.Variables created under the scope.