Class

MPSRNNMatrixTrainingState

A class that holds data from a forward pass to be used in a backward pass.

Declaration

@interface MPSRNNMatrixTrainingState : MPSState

Relationships

Inherits From

See Also

Recurrent Neural Networks

MPSRNNImageInferenceLayer

A recurrent neural network layer for inference on Metal Performance Shaders images.

MPSRNNMatrixInferenceLayer

A recurrent neural network layer for inference on Metal Performance Shaders matrices.

MPSRNNSingleGateDescriptor

A description of a simple recurrent block or layer.

MPSGRUDescriptor

A description of a gated recurrent unit block or layer.

MPSLSTMDescriptor

A description of a long short-term memory block or layer.

MPSRNNSequenceDirection

Directions that a sequence of inputs can be processed by a recurrent neural network layer.

MPSRNNMatrixTrainingLayer

A layer for training recurrent neural networks on Metal Performance Shaders matrices.