@class NSObject;
@protocol OS_dispatch_semaphore;

@interface MPSNNOptimizerAdam : MPSNNOptimizer {
    NSObject<OS_dispatch_semaphore> *_timeStepSemaphore;
}

@property (readonly, nonatomic) double beta1;
@property (readonly, nonatomic) double beta2;
@property (readonly, nonatomic) float epsilon;
@property (nonatomic) unsigned long long timeStep;

+ (const struct MPSLibraryInfo { int x0; unsigned int x1; char *x2; void /* function */ *x3; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x4; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x5; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x6; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x7; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x8; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x9; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x10; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x11; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x12; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x13; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x14; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x15; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x16; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x17; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x18; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x19; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x20; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x21; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x22; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x23; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x24; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x25; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x26; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x27; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x28; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x29; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x30; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x31; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x32; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x33; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x34; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x35; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x36; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x37; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x38; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x39; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x40; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x41; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x42; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x43; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x44; struct MPSDeviceSpecificInfo { struct MPSKernelInfo *x0; void /* function */ *x1; unsigned long long x2; } x45; } *)libraryInfo:(void *)a0;

- (id)copyWithZone:(struct _NSZone { } *)a0 device:(id)a1;
- (id)initWithCoder:(id)a0 device:(id)a1;
- (void)dealloc;
- (void)encodeWithCoder:(id)a0;
- (id)debugDescription;
- (void)encodeToCommandBuffer:(id)a0 inputGradientVector:(id)a1 inputValuesVector:(id)a2 inputMomentumVector:(id)a3 inputVelocityVector:(id)a4 resultValuesVector:(id)a5;
- (id)initWithDevice:(id)a0 beta1:(double)a1 beta2:(double)a2 epsilon:(float)a3 timeStep:(unsigned long long)a4 optimizerDescriptor:(id)a5;
- (id)initWithDevice:(id)a0 learningRate:(float)a1;
- (void)encodeToCommandBuffer:(id)a0 convolutionGradientState:(id)a1 convolutionSourceState:(id)a2 inputMomentumVectors:(id)a3 inputVelocityVectors:(id)a4 resultState:(id)a5;
- (void)encodeToCommandBuffer:(id)a0 batchNormalizationState:(id)a1 inputMomentumVectors:(id)a2 inputVelocityVectors:(id)a3 resultState:(id)a4;
- (void)encodeToCommandBuffer:(id)a0 batchNormalizationGradientState:(id)a1 batchNormalizationSourceState:(id)a2 inputMomentumVectors:(id)a3 inputVelocityVectors:(id)a4 maximumVelocityVectors:(id)a5 resultState:(id)a6;
- (void)encodeToCommandBuffer:(id)a0 batchNormalizationGradientState:(id)a1 batchNormalizationSourceState:(id)a2 inputMomentumVectors:(id)a3 inputVelocityVectors:(id)a4 resultState:(id)a5;
- (void)encodeToCommandBuffer:(id)a0 batchNormalizationState:(id)a1 inputMomentumVectors:(id)a2 inputVelocityVectors:(id)a3 maximumVelocityVectors:(id)a4 resultState:(id)a5;
- (void)encodeToCommandBuffer:(id)a0 convolutionGradientState:(id)a1 convolutionSourceState:(id)a2 inputMomentumVectors:(id)a3 inputVelocityVectors:(id)a4 maximumVelocityVectors:(id)a5 resultState:(id)a6;
- (void)encodeToCommandBuffer:(id)a0 inputGradientMatrix:(id)a1 inputValuesMatrix:(id)a2 inputMomentumMatrix:(id)a3 inputVelocityMatrix:(id)a4 maximumVelocityMatrix:(id)a5 resultValuesMatrix:(id)a6;
- (void)encodeToCommandBuffer:(id)a0 inputGradientMatrix:(id)a1 inputValuesMatrix:(id)a2 inputMomentumMatrix:(id)a3 inputVelocityMatrix:(id)a4 resultValuesMatrix:(id)a5;
- (void)encodeToCommandBuffer:(id)a0 inputGradientVector:(id)a1 inputValuesVector:(id)a2 inputMomentumVector:(id)a3 inputVelocityVector:(id)a4 maximumVelocityVector:(id)a5 resultValuesVector:(id)a6;
- (id)initWithDevice:(id)a0 optimizerDescriptor:(id)a1;

@end