layoutmap - Robuta Search

https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/dtensor/experimental/LayoutMap tf.keras.dtensor.experimental.LayoutMap | TensorFlow v2.9.3 A dict-like object that maps string to Layout instances. tf keras dtensor experimental layoutmap