Package org.robovm.apple.metalps

Class MPSRNNMatrixTrainingLayer

java.lang.Object

org.robovm.rt.bro.NativeObject

org.robovm.objc.ObjCObject

org.robovm.apple.foundation.NSObject

org.robovm.apple.metalps.MPSKernel

org.robovm.apple.metalps.MPSRNNMatrixTrainingLayer

All Implemented Interfaces:

NSCoding, NSObjectProtocol, NSSecureCoding, org.robovm.objc.ObjCProtocol

public class MPSRNNMatrixTrainingLayer
extends MPSKernel

Since:

Available in iOS 12.0 and later.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MPSRNNMatrixTrainingLayer.MPSRNNMatrixTrainingLayerPtr

Nested classes/interfaces inherited from class org.robovm.apple.metalps.MPSKernel

MPSKernel.MPSKernelPtr

Nested classes/interfaces inherited from class org.robovm.apple.foundation.NSObject

NSObject.Handle, NSObject.Marshaler, NSObject.NoRetainMarshaler, NSObject.NSKeyValueObserver, NSObject.NSObjectPtr, NSObject.SkipInit

Nested classes/interfaces inherited from class org.robovm.objc.ObjCObject

org.robovm.objc.ObjCObject.ObjCObjectPtr, org.robovm.objc.ObjCObject.Super

Field Summary

Fields inherited from class org.robovm.apple.foundation.NSObject

FLAG_NO_RETAIN

Fields inherited from class org.robovm.objc.ObjCObject

customClass

Constructor Summary

Constructors

Modifier	Constructor	Description
	MPSRNNMatrixTrainingLayer()
	MPSRNNMatrixTrainingLayer(NSCoder decoder)
	MPSRNNMatrixTrainingLayer(NSCoder decoder, MTLDevice device)
protected	MPSRNNMatrixTrainingLayer(NSObject.Handle h, long handle)
protected	MPSRNNMatrixTrainingLayer(NSObject.SkipInit skipInit)
	MPSRNNMatrixTrainingLayer(MTLDevice device, MPSRNNDescriptor rnnDescriptor, NSMutableArray<MPSMatrix> trainableWeights)

Method Summary

Modifier and Type	Method	Description
MPSRNNMatrixTrainingLayer	copy(NSZone zone, MTLDevice device)
void	createTemporaryWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut, MPSDataType dataType, MTLCommandBuffer commandBuffer)
void	createWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut, MPSDataType dataType)
void	createWeightMatrices(NSMutableArray<MPSMatrix> matricesOut)
void	encodeCopyWeightsToCommandBuffer(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> weights, MPSRNNMatrixId matrixId, MPSMatrix matrix, boolean copyFromWeightsToMatrix, MTLOrigin matrixOffset)
void	encodeForwardSequenceToCommandBuffer(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> sourceMatrices, NSArray<MPSMatrix> destinationMatrices, NSMutableArray<MPSRNNMatrixTrainingState> trainingStates, NSArray<MPSMatrix> weights)
void	encodeForwardSequenceToCommandBuffer(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> sourceMatrices, org.robovm.rt.bro.ptr.MachineSizedUIntPtr sourceOffsets, NSArray<MPSMatrix> destinationMatrices, org.robovm.rt.bro.ptr.MachineSizedUIntPtr destinationOffsets, NSMutableArray<MPSRNNMatrixTrainingState> trainingStates, MPSRNNRecurrentMatrixState recurrentInputState, NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates, NSArray<MPSMatrix> weights)
void	encodeGradientSequenceToCommandBuffer(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> forwardSources, NSArray<MPSMatrix> sourceGradients, NSArray<MPSMatrix> destinationGradients, NSArray<MPSMatrix> weightGradients, NSArray<MPSRNNMatrixTrainingState> trainingStates, NSArray<MPSMatrix> weights)
void	encodeGradientSequenceToCommandBuffer(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> forwardSources, org.robovm.rt.bro.ptr.MachineSizedUIntPtr forwardSourceOffsets, NSArray<MPSMatrix> sourceGradients, org.robovm.rt.bro.ptr.MachineSizedUIntPtr sourceGradientOffsets, NSArray<MPSMatrix> destinationGradients, org.robovm.rt.bro.ptr.MachineSizedUIntPtr destinationOffsets, NSArray<MPSMatrix> weightGradients, NSArray<MPSRNNMatrixTrainingState> trainingStates, MPSRNNRecurrentMatrixState recurrentInputState, NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates, NSArray<MPSMatrix> weights)
long	getInputFeatureChannels()
long	getOutputFeatureChannels()
protected long	init(NSCoder decoder, MTLDevice device)
protected long	init(MTLDevice device, MPSRNNDescriptor rnnDescriptor, NSMutableArray<MPSMatrix> trainableWeights)
boolean	isAccumulateWeightGradients()
boolean	isRecurrentOutputIsTemporary()
boolean	isStoreAllIntermediateStates()
boolean	isTrainingStateIsTemporary()
void	setAccumulateWeightGradients(boolean v)
void	setRecurrentOutputIsTemporary(boolean v)
void	setStoreAllIntermediateStates(boolean v)
void	setTrainingStateIsTemporary(boolean v)

Methods inherited from class org.robovm.apple.metalps.MPSKernel

encode, getDevice, getLabel, getOptions, init, init, setLabel, setOptions, supportsSecureCoding

Methods inherited from class org.robovm.apple.foundation.NSObject

addKeyValueObserver, addKeyValueObserver, afterMarshaled, alloc, autorelease, autorelease, awakeFromNib, conformsToProtocol, copy, description, didChangeValue, didChangeValue, didChangeValues, doDispose, equals, forceSkipInit, getAutoContentAccessingProxy, getClassForCoder, getClassForKeyedArchiver, getKeyValueCoder, getObservationInfo, hash, hashCode, init, isEqual, isKindOfClass, isMemberOfClass, mutableCopy, performSelector, performSelector, performSelector, performSelector, performSelector, performSelector, performSelector, performSelectorInBackground, performSelectorOnMainThread, performSelectorOnMainThread, performSelectorV, performSelectorV, performSelectorV, release, release, removeKeyValueObserver, removeKeyValueObservers, respondsToSelector, retain, retain, retainCount, setObservationInfo, toString, willChangeValue, willChangeValue, willChangeValues

Methods inherited from class org.robovm.objc.ObjCObject

addStrongRef, dispose, dispose, finalize, getAssociatedObject, getObjCClass, getPeerObject, getSuper, initObject, logRetainRelease, removeStrongRef, retainCustomObjectFromCb, setAssociatedObject, toObjCObject, toObjCObject, updateStrongRef

Methods inherited from class org.robovm.rt.bro.NativeObject

as, getHandle, setHandle

Methods inherited from class java.lang.Object

clone, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

MPSRNNMatrixTrainingLayer

public MPSRNNMatrixTrainingLayer()

MPSRNNMatrixTrainingLayer

protected MPSRNNMatrixTrainingLayer(NSObject.Handle h,
                                    long handle)

MPSRNNMatrixTrainingLayer

protected MPSRNNMatrixTrainingLayer(NSObject.SkipInit skipInit)

MPSRNNMatrixTrainingLayer

public MPSRNNMatrixTrainingLayer(MTLDevice device,
                                 MPSRNNDescriptor rnnDescriptor,
                                 NSMutableArray<MPSMatrix> trainableWeights)

Since:

Available in iOS 12.0 and later.

MPSRNNMatrixTrainingLayer

public MPSRNNMatrixTrainingLayer(NSCoder decoder,
                                 MTLDevice device)

Since:

Available in iOS 11.0 and later.

MPSRNNMatrixTrainingLayer

public MPSRNNMatrixTrainingLayer(NSCoder decoder)

Method Detail

getInputFeatureChannels

public long getInputFeatureChannels()

getOutputFeatureChannels

public long getOutputFeatureChannels()

isStoreAllIntermediateStates

public boolean isStoreAllIntermediateStates()

setStoreAllIntermediateStates

public void setStoreAllIntermediateStates(boolean v)

isRecurrentOutputIsTemporary

public boolean isRecurrentOutputIsTemporary()

setRecurrentOutputIsTemporary

public void setRecurrentOutputIsTemporary(boolean v)

isTrainingStateIsTemporary

public boolean isTrainingStateIsTemporary()

setTrainingStateIsTemporary

public void setTrainingStateIsTemporary(boolean v)

isAccumulateWeightGradients

public boolean isAccumulateWeightGradients()

setAccumulateWeightGradients

public void setAccumulateWeightGradients(boolean v)

init

protected long init(MTLDevice device,
                    MPSRNNDescriptor rnnDescriptor,
                    NSMutableArray<MPSMatrix> trainableWeights)

Since:

Available in iOS 12.0 and later.

createWeightGradientMatrices

public void createWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut,
                                         MPSDataType dataType)

Since:

Available in iOS 12.0 and later.

createTemporaryWeightGradientMatrices

public void createTemporaryWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut,
                                                  MPSDataType dataType,
                                                  MTLCommandBuffer commandBuffer)

Since:

Available in iOS 12.0 and later.

createWeightMatrices

public void createWeightMatrices(NSMutableArray<MPSMatrix> matricesOut)

Since:

Available in iOS 12.0 and later.

encodeCopyWeightsToCommandBuffer

public void encodeCopyWeightsToCommandBuffer(MTLCommandBuffer commandBuffer,
                                             NSArray<MPSMatrix> weights,
                                             MPSRNNMatrixId matrixId,
                                             MPSMatrix matrix,
                                             boolean copyFromWeightsToMatrix,
                                             MTLOrigin matrixOffset)

encodeForwardSequenceToCommandBuffer

public void encodeForwardSequenceToCommandBuffer(MTLCommandBuffer commandBuffer,
                                                 NSArray<MPSMatrix> sourceMatrices,
                                                 org.robovm.rt.bro.ptr.MachineSizedUIntPtr sourceOffsets,
                                                 NSArray<MPSMatrix> destinationMatrices,
                                                 org.robovm.rt.bro.ptr.MachineSizedUIntPtr destinationOffsets,
                                                 NSMutableArray<MPSRNNMatrixTrainingState> trainingStates,
                                                 MPSRNNRecurrentMatrixState recurrentInputState,
                                                 NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
                                                 NSArray<MPSMatrix> weights)

encodeForwardSequenceToCommandBuffer

public void encodeForwardSequenceToCommandBuffer(MTLCommandBuffer commandBuffer,
                                                 NSArray<MPSMatrix> sourceMatrices,
                                                 NSArray<MPSMatrix> destinationMatrices,
                                                 NSMutableArray<MPSRNNMatrixTrainingState> trainingStates,
                                                 NSArray<MPSMatrix> weights)

encodeGradientSequenceToCommandBuffer

public void encodeGradientSequenceToCommandBuffer(MTLCommandBuffer commandBuffer,
                                                  NSArray<MPSMatrix> forwardSources,
                                                  org.robovm.rt.bro.ptr.MachineSizedUIntPtr forwardSourceOffsets,
                                                  NSArray<MPSMatrix> sourceGradients,
                                                  org.robovm.rt.bro.ptr.MachineSizedUIntPtr sourceGradientOffsets,
                                                  NSArray<MPSMatrix> destinationGradients,
                                                  org.robovm.rt.bro.ptr.MachineSizedUIntPtr destinationOffsets,
                                                  NSArray<MPSMatrix> weightGradients,
                                                  NSArray<MPSRNNMatrixTrainingState> trainingStates,
                                                  MPSRNNRecurrentMatrixState recurrentInputState,
                                                  NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
                                                  NSArray<MPSMatrix> weights)

encodeGradientSequenceToCommandBuffer

public void encodeGradientSequenceToCommandBuffer(MTLCommandBuffer commandBuffer,
                                                  NSArray<MPSMatrix> forwardSources,
                                                  NSArray<MPSMatrix> sourceGradients,
                                                  NSArray<MPSMatrix> destinationGradients,
                                                  NSArray<MPSMatrix> weightGradients,
                                                  NSArray<MPSRNNMatrixTrainingState> trainingStates,
                                                  NSArray<MPSMatrix> weights)

init

protected long init(NSCoder decoder,
                    MTLDevice device)

Overrides:

init in class MPSKernel

Since:

Available in iOS 11.0 and later.

copy

public MPSRNNMatrixTrainingLayer copy(NSZone zone,
                                      MTLDevice device)

Overrides:

copy in class MPSKernel