LostTech.TensorFlow : API Documentation

Type CentralStorageStrategy

Namespace tensorflow.distribute.experimental

Parent Strategy

Interfaces ICentralStorageStrategy

A one-machine strategy that puts all variables on a single device.

Variables are assigned to local CPU or the only GPU. If there is more than one GPU, compute operations (other than variable update operations) will be replicated across all GPUs.

For Example: ``` strategy = tf.distribute.experimental.CentralStorageStrategy() # Create a dataset ds = tf.data.Dataset.range(5).batch(2) # Distribute that dataset dist_dataset = strategy.experimental_distribute_dataset(ds)

with strategy.scope(): @tf.function def train_step(val): return val + 1

# Iterate over the distributed dataset for x in dist_dataset: # process dataset elements strategy.experimental_run_v2(train_step, args=(x,)) ```


Public properties

object extended get;

object extended_dyn get;

int num_replicas_in_sync get;

object num_replicas_in_sync_dyn get;

object PythonObject get;