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 RMSprop
: DTensor specific optimizers.
class SGD
: DTensor specific optimizers.