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