Package org.robovm.apple.metalps
Class MPSNNOptimizerAdam
- 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.MPSNNOptimizer
-
- org.robovm.apple.metalps.MPSNNOptimizerAdam
-
- All Implemented Interfaces:
NSCoding
,NSObjectProtocol
,NSSecureCoding
,org.robovm.objc.ObjCProtocol
public class MPSNNOptimizerAdam extends MPSNNOptimizer
- Since:
- Available in iOS 12.0 and later.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
MPSNNOptimizerAdam.MPSNNOptimizerAdamPtr
-
Nested classes/interfaces inherited from class org.robovm.apple.metalps.MPSNNOptimizer
MPSNNOptimizer.MPSNNOptimizerPtr
-
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
-
-
Field Summary
-
Fields inherited from class org.robovm.apple.foundation.NSObject
FLAG_NO_RETAIN
-
-
Constructor Summary
Constructors Modifier Constructor Description MPSNNOptimizerAdam()
MPSNNOptimizerAdam(NSCoder decoder)
MPSNNOptimizerAdam(NSCoder decoder, MTLDevice device)
protected
MPSNNOptimizerAdam(NSObject.Handle h, long handle)
protected
MPSNNOptimizerAdam(NSObject.SkipInit skipInit)
MPSNNOptimizerAdam(MTLDevice device, double beta1, double beta2, float epsilon, long timeStep, MPSNNOptimizerDescriptor optimizerDescriptor)
MPSNNOptimizerAdam(MTLDevice device, float learningRate)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNBatchNormalizationState batchNormalizationState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNNormalizationGammaAndBetaState resultState)
void
encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNBatchNormalizationState batchNormalizationGradientState, MPSCNNBatchNormalizationState batchNormalizationSourceState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNNormalizationGammaAndBetaState resultState)
void
encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNConvolutionGradientState convolutionGradientState, MPSCNNConvolutionWeightsAndBiasesState convolutionSourceState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNConvolutionWeightsAndBiasesState resultState)
void
encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSVector inputGradientVector, MPSVector inputValuesVector, MPSVector inputMomentumVector, MPSVector inputVelocityVector, MPSVector resultValuesVector)
double
getBeta1()
double
getBeta2()
float
getEpsilon()
long
getTimeStep()
protected long
init(MTLDevice device, double beta1, double beta2, float epsilon, long timeStep, MPSNNOptimizerDescriptor optimizerDescriptor)
protected long
init(MTLDevice device, float learningRate)
void
setTimeStep(long v)
-
Methods inherited from class org.robovm.apple.metalps.MPSNNOptimizer
getGradientClipMax, getGradientClipMin, getGradientRescale, getLearningRate, getRegularizationScale, getRegularizationType, isApplyGradientClipping, setApplyGradientClipping, setLearningRate
-
Methods inherited from class org.robovm.apple.metalps.MPSKernel
copy, encode, getDevice, getLabel, getOptions, init, 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
-
-
-
-
Constructor Detail
-
MPSNNOptimizerAdam
public MPSNNOptimizerAdam()
-
MPSNNOptimizerAdam
protected MPSNNOptimizerAdam(NSObject.Handle h, long handle)
-
MPSNNOptimizerAdam
protected MPSNNOptimizerAdam(NSObject.SkipInit skipInit)
-
MPSNNOptimizerAdam
public MPSNNOptimizerAdam(MTLDevice device, float learningRate)
-
MPSNNOptimizerAdam
public MPSNNOptimizerAdam(MTLDevice device, double beta1, double beta2, float epsilon, long timeStep, MPSNNOptimizerDescriptor optimizerDescriptor)
-
MPSNNOptimizerAdam
public MPSNNOptimizerAdam(NSCoder decoder)
-
-
Method Detail
-
getBeta1
public double getBeta1()
-
getBeta2
public double getBeta2()
-
getEpsilon
public float getEpsilon()
-
getTimeStep
public long getTimeStep()
-
setTimeStep
public void setTimeStep(long v)
-
init
protected long init(MTLDevice device, float learningRate)
-
init
protected long init(MTLDevice device, double beta1, double beta2, float epsilon, long timeStep, MPSNNOptimizerDescriptor optimizerDescriptor)
-
encodeToCommandBuffer
public void encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSVector inputGradientVector, MPSVector inputValuesVector, MPSVector inputMomentumVector, MPSVector inputVelocityVector, MPSVector resultValuesVector)
-
encodeToCommandBuffer
public void encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNConvolutionGradientState convolutionGradientState, MPSCNNConvolutionWeightsAndBiasesState convolutionSourceState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNConvolutionWeightsAndBiasesState resultState)
-
encodeToCommandBuffer
public void encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNBatchNormalizationState batchNormalizationState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNNormalizationGammaAndBetaState resultState)
-
encodeToCommandBuffer
public void encodeToCommandBuffer(MTLCommandBuffer commandBuffer, MPSCNNBatchNormalizationState batchNormalizationGradientState, MPSCNNBatchNormalizationState batchNormalizationSourceState, NSArray<MPSVector> inputMomentumVectors, NSArray<MPSVector> inputVelocityVectors, MPSCNNNormalizationGammaAndBetaState resultState)
-
-