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.  

    
{'path': 'public/docs/tensorflow~2.9/keras/dtensor/experimental.html', 'source': 'DevDocs.IO'}