https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/DVariable?authuser=19
tf.experimental.dtensor.DVariable | TensorFlow v2.16.1
A replacement for tf.Variable which follows initial value placement.
tfexperimentaldtensordvariabletensorflow
https://www.tensorflow.org/versions/r2.12/api_docs/python/tf/experimental/dtensor/Mesh
tf.experimental.dtensor.Mesh | TensorFlow v2.12.1
Represents a Mesh configuration over a certain list of Mesh Dimensions.
tfexperimentaldtensormeshtensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/Mesh?authuser=19
tf.experimental.dtensor.Mesh | TensorFlow v2.16.1
Represents a Mesh configuration over a certain list of Mesh Dimensions.
tfexperimentaldtensormeshtensorflow
https://www.tensorflow.org/versions/r2.11/api_docs/python/tf/keras/dtensor/experimental/layout_map_scope
tf.keras.dtensor.experimental.layout_map_scope | TensorFlow v2.11.1
Apply the layout to all the tf.Variables created under the scope. (deprecated)
tfkerasdtensorexperimentallayout
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.
tfkerasdtensorexperimentallayoutmap
https://www.tensorflow.org/versions/r2.13/api_docs/python/tf/experimental/dtensor/Mesh?authuser=1
tf.experimental.dtensor.Mesh | TensorFlow v2.13.1
Represents a Mesh configuration over a certain list of Mesh Dimensions.
tfexperimentaldtensormeshtensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor?authuser=00
Module: tf.experimental.dtensor | TensorFlow v2.16.1
Public API for tf._api.v2.experimental.dtensor namespace
moduletfexperimentaldtensortensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/Layout?authuser=00
tf.experimental.dtensor.Layout | TensorFlow v2.16.1
Represents the layout information of a DTensor.
tfexperimentaldtensorlayouttensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/pack?authuser=19
tf.experimental.dtensor.pack | TensorFlow v2.16.1
Packs tf.Tensor components into a DTensor.
tfexperimentaldtensorpacktensorflow
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/experimental/dtensor/enable_save_as_bf16
tf.experimental.dtensor.enable_save_as_bf16 | TensorFlow v2.9.3
Allows float32 DVariables to be checkpointed and restored as bfloat16.
save astfexperimentaldtensorenable
https://www.tensorflow.org/versions/r2.14/api_docs/python/tf/experimental/dtensor/call_with_layout
tf.experimental.dtensor.call_with_layout | TensorFlow v2.14.0
Calls a function in the DTensor device scope if layout is not None.
with layouttfexperimentaldtensorcall
https://www.tensorflow.org/versions/r2.13/api_docs/python/tf/experimental/dtensor/name_based_save
tf.experimental.dtensor.name_based_save | TensorFlow v2.13.1
Saves name based Tensor into a Checkpoint.
tfexperimentaldtensornamebased
https://www.tensorflow.org/versions/r2.13/api_docs/python/tf/experimental/dtensor/DVariable
tf.experimental.dtensor.DVariable | TensorFlow v2.13.1
A replacement for tf.Variable which follows initial value placement.
tfexperimentaldtensordvariabletensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/create_mesh?authuser=00
tf.experimental.dtensor.create_mesh | TensorFlow v2.16.1
Creates a single-client mesh.
tfexperimentaldtensorcreatemesh
https://www.tensorflow.org/versions/r2.12/api_docs/python/tf/experimental/dtensor/Layout
tf.experimental.dtensor.Layout | TensorFlow v2.12.1
Represents the layout information of a DTensor.
tfexperimentaldtensorlayouttensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/call_with_layout?authuser=1
tf.experimental.dtensor.call_with_layout | TensorFlow v2.16.1
Calls a function in the DTensor device scope if layout is not None.
with layouttfexperimentaldtensorcall
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/Mesh?authuser=00
tf.experimental.dtensor.Mesh | TensorFlow v2.16.1
Represents a Mesh configuration over a certain list of Mesh Dimensions.
tfexperimentaldtensormeshtensorflow
https://www.tensorflow.org/versions/r2.13/api_docs/python/tf/experimental/dtensor/relayout
tf.experimental.dtensor.relayout | TensorFlow v2.13.1
Changes the layout of tensor.
tfexperimentaldtensorrelayouttensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/DVariable?authuser=00
tf.experimental.dtensor.DVariable | TensorFlow v2.16.1
A replacement for tf.Variable which follows initial value placement.
tfexperimentaldtensordvariabletensorflow
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/create_mesh?authuser=19
tf.experimental.dtensor.create_mesh | TensorFlow v2.16.1
Creates a single-client mesh.
tfexperimentaldtensorcreatemesh
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/name_based_restore?authuser=1
tf.experimental.dtensor.name_based_restore | TensorFlow v2.16.1
Restores from checkpoint_prefix to name based DTensors.
tfexperimentaldtensornamebased
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/num_clients?authuser=1
tf.experimental.dtensor.num_clients | TensorFlow v2.16.1
Returns the number of clients in this DTensor cluster.
tfexperimentaldtensornumclients
https://www.tensorflow.org/versions/r2.12/api_docs/python/tf/experimental/dtensor/name_based_restore
tf.experimental.dtensor.name_based_restore | TensorFlow v2.12.1
Restores from checkpoint_prefix to name based DTensors.
tfexperimentaldtensornamebased
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/experimental/dtensor/check_layout
tf.experimental.dtensor.check_layout | TensorFlow v2.9.3
Asserts that the layout of the DTensor is layout.
tfexperimentaldtensorchecklayout
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/create_distributed_mesh?authuser=00
tf.experimental.dtensor.create_distributed_mesh | TensorFlow v2.16.1
Creates a distributed mesh.
tfexperimentaldtensorcreatedistributed