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, 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

ObjCObject.ObjCObjectPtr, 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 coder)
	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	encodeCopyWeights(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> weights, MPSRNNMatrixId matrixId, MPSMatrix matrix, boolean copyFromWeightsToMatrix, MTLOrigin matrixOffset)
void	encodeForwardSequence(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> sourceMatrices, NSArray<MPSMatrix> destinationMatrices, NSMutableArray<MPSRNNMatrixTrainingState> trainingStates, NSArray<MPSMatrix> weights)
void	encodeForwardSequence(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> sourceMatrices, MachineSizedUIntPtr sourceOffsets, NSArray<MPSMatrix> destinationMatrices, MachineSizedUIntPtr destinationOffsets, NSMutableArray<MPSRNNMatrixTrainingState> trainingStates, MPSRNNRecurrentMatrixState recurrentInputState, NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates, NSArray<MPSMatrix> weights)
void	encodeGradientSequence(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> forwardSources, NSArray<MPSMatrix> sourceGradients, NSArray<MPSMatrix> destinationGradients, NSArray<MPSMatrix> weightGradients, NSArray<MPSRNNMatrixTrainingState> trainingStates, NSArray<MPSMatrix> weights)
void	encodeGradientSequence(MTLCommandBuffer commandBuffer, NSArray<MPSMatrix> forwardSources, MachineSizedUIntPtr forwardSourceOffsets, NSArray<MPSMatrix> sourceGradients, MachineSizedUIntPtr sourceGradientOffsets, NSArray<MPSMatrix> destinationGradients, 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)
static boolean	supportsSecureCoding()

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

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

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 Details

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 coder)

Method Details

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)

supportsSecureCoding

public static boolean supportsSecureCoding()

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.

encodeCopyWeights

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

encodeForwardSequence

public void encodeForwardSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceMatrices,
MachineSizedUIntPtr sourceOffsets,
NSArray<MPSMatrix> destinationMatrices,
MachineSizedUIntPtr destinationOffsets,
NSMutableArray<MPSRNNMatrixTrainingState> trainingStates,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
NSArray<MPSMatrix> weights)

encodeForwardSequence

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

encodeGradientSequence

public void encodeGradientSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> forwardSources,
MachineSizedUIntPtr forwardSourceOffsets,
NSArray<MPSMatrix> sourceGradients,
MachineSizedUIntPtr sourceGradientOffsets,
NSArray<MPSMatrix> destinationGradients,
MachineSizedUIntPtr destinationOffsets,
NSArray<MPSMatrix> weightGradients,
NSArray<MPSRNNMatrixTrainingState> trainingStates,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
NSArray<MPSMatrix> weights)

encodeGradientSequence

public void encodeGradientSequence(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