Module: tf.keras.dtensor.experimental.optimizers

Public API for tf.keras.dtensor.experimental.optimizers namespace.

Classes

class Adadelta: DTensor specific optimizers.

class Adagrad: DTensor specific optimizers.

class Adam: DTensor specific optimizers.

class AdamW: DTensor specific optimizers.

class RMSprop: DTensor specific optimizers.

class SGD: DTensor specific optimizers.