MPSRNNMatrixTrainingLayer
A layer for training recurrent neural networks on Metal Performance Shaders matrices.
Declaration
class MPSRNNMatrixTrainingLayerTopics
Initializers
Instance Properties
accumulateWeightGradientsinputFeatureChannelsoutputFeatureChannelsrecurrentOutputIsTemporarystoreAllIntermediateStatestrainingStateIsTemporary
Instance Methods
copy(with:device:)createTemporaryWeightGradientMatrices(_:dataType:commandBuffer:)createWeightGradientMatrices(_:dataType:)createWeightMatrices(_:)encodeCopyWeights(commandBuffer:weights:matrixId:matrix:copyFromWeightsToMatrix:matrixOffset:)encodeForwardSequence(commandBuffer:sourceMatrices:destinationMatrices:trainingStates:weights:)encodeForwardSequence(commandBuffer:sourceMatrices:sourceOffsets:destinationMatrices:destinationOffsets:trainingStates:recurrentInputState:recurrentOutputStates:weights:)encodeGradientSequence(commandBuffer:forwardSources:forwardSourceOffsets:sourceGradients:sourceOffsets:destinationGradients:destinationOffsets:weightGradients:trainingStates:recurrentInputState:recurrentOutputStates:weights:)encodeGradientSequence(commandBuffer:forwardSources:sourceGradients:destinationGradients:weightGradients:trainingStates:weights:)