tf.layers.FeedForwardNetworkV2 module

tf.layers.FeedForwardNetworkV2 module

class tf.layers.FeedForwardNetworkV2.FeedForwardNetworkV2(*args: Any, **kwargs: Any)

Bases: modelzoo.common.tf.layers.BaseLayer.BaseLayer

Implement a feed forward network as used in the T5 model.

call(inputs, training=True, **kwargs)