@class NSString, NSArray, NSData;
@interface MLCDeviceCPU : NSObject <MLCLayerOperations, MLCEngineDispatch, MLComputeEngineOptimizerUpdate, NSCopying, MLComputeEngineControl>
@property (readonly) unsigned long long hash;
@property (readonly) Class superclass;
@property (readonly, copy) NSString *description;
@property (readonly, copy) NSString *debugDescription;
@property (readonly) unsigned long long hash;
@property (readonly) Class superclass;
@property (readonly, copy) NSString *description;
@property (readonly, copy) NSString *debugDescription;
@property (readonly) unsigned long long hash;
@property (readonly) Class superclass;
@property (readonly, copy) NSString *description;
@property (readonly, copy) NSString *debugDescription;
@property (readonly, nonatomic) NSArray *deviceList;
@property (readonly, nonatomic) int deviceType;
@property (retain, nonatomic) NSData *deviceHeap;
@property (readonly) unsigned long long hash;
@property (readonly) Class superclass;
@property (readonly, copy) NSString *description;
@property (readonly, copy) NSString *debugDescription;
- (void).cxx_destruct;
- (id)initWithType:(int)a0;
- (id)copyWithZone:(struct _NSZone { } *)a0;
- (unsigned long long)numDevices;
- (void)dispatchForwardEmbeddingLayer:(id)a0 weight:(id)a1 sourceTensor:(id)a2 resultTensor:(id)a3;
- (id)gammaGradients:(id)a0;
- (id)gramMatrixLayerWithScaleFactor:(float)a0;
- (id)mhaBiasGradient:(id)a0 withSize:(unsigned long long)a1 index:(unsigned long long)a2;
- (void)signalNextEvent;
- (void)dispatchForwardSplitLayer:(id)a0 sourceTensor:(id)a1 resultTensors:(id)a2 forConcat:(BOOL)a3;
- (void)accumulateParams:(void **)a0 gradients:(void **)a1 accumulators:(void **)a2 numOfParameters:(unsigned long long)a3 inArrayOfParams:(id)a4;
- (void)allocateDeviceDataForGlobalNormClippingWithOptimizer:(id)a0 updatableParameterCount:(unsigned long long)a1;
- (void)allocateDeviceMemoryForTensor:(id)a0;
- (void)allocateParameterGradientsForDeviceOps:(id)a0 parameters:(id)a1;
- (void)allocateResultTensor:(id)a0;
- (unsigned long long)allocatedDeviceDataSizeForTraining:(BOOL)a0 layer:(id)a1;
- (id)arithmeticLayerWithOperation:(int)a0 activationDescriptor:(id)a1;
- (id)batchNormalizationLayerWithChannelCount:(unsigned long long)a0 mean:(id)a1 variance:(id)a2 beta:(id)a3 gamma:(id)a4 varianceEpsilon:(float)a5 momentum:(float)a6;
- (id)betaGradients:(id)a0;
- (id)biasesGradients:(id)a0;
- (void)broadcastTensor:(id)a0;
- (void)broadcastTensor:(id)a0 batchSize:(unsigned long long)a1;
- (void)commitAndWaitForCompletion:(id /* block */)a0;
- (void)commitAndWaitForCompletion:(id /* block */)a0 enableProfiling:(BOOL)a1 graphExecutionTime:(id)a2 graphResultTensor:(id)a3;
- (void)commitWithCompletionHandler:(id /* block */)a0 enableProfiling:(BOOL)a1 graphExecutionTime:(id)a2 graphResultTensor:(id)a3;
- (void)commitWithProfiling:(BOOL)a0 graphExecutionTime:(id)a1;
- (id)compareLayerWithOperation:(int)a0;
- (BOOL)compileLayerDeviceOps:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2;
- (BOOL)compileLayerDeviceOps:(id)a0 sourceTensors:(id)a1 resultTensors:(id)a2;
- (void)convertUpdatesToTensorDataForLayer:(id)a0;
- (void)convertUpdatesToTensorDataForTensorParameters:(id)a0;
- (id)convolutionLayerWithDescriptor:(id)a0 weights:(id)a1 biasTerms:(id)a2;
- (id)convolutionTransposeLayerWithDescriptor:(id)a0 weights:(id)a1 biasTerms:(id)a2;
- (void)copyBiasFromBNNSToMLCTensor:(id)a0 biasTensors:(id)a1 baseGateIndex:(unsigned long long)a2 lstmDescriptor:(id)a3;
- (void)createBNNSDesc2DForLSTMWeights:(id)a0 tensorParameters:(id)a1 deviceOptimizers:(id)a2 isInputWeight:(BOOL)a3;
- (id)createOptimizerDeviceDataForTensor:(id)a0 optimizer:(id)a1 optimizerData:(id)a2 isVector:(BOOL)a3;
- (void)deallocateDeviceMemoryForTensor:(id)a0;
- (unsigned long long)deviceMemorySizeForTensor:(id)a0;
- (unsigned long long)deviceMemorySizeForTensor:(id)a0 batchSize:(unsigned long long)a1;
- (void)dispatchBroadcastTensor:(id)a0;
- (void)dispatchBroadcastTensor:(id)a0 batchSize:(unsigned long long)a1;
- (void)dispatchForwardAndGradientLossLayer:(id)a0 sourceTensor:(id)a1 labelsTensor:(id)a2 labelsTensorStride:(unsigned long long)a3 weightsTensor:(id)a4 resultTensor:(id)a5 resultGradientTensor:(id)a6;
- (void)dispatchForwardCompareLayer:(id)a0 sourceTensor:(id)a1 secondaryTensor:(id)a2 resultTensor:(id)a3 compareOp:(int)a4 forTraining:(BOOL)a5;
- (void)dispatchForwardConcatLayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2;
- (void)dispatchForwardFullyConnectedLayer:(id)a0 sourceTensor:(id)a1 resultTensor:(id)a2 forTraining:(BOOL)a3;
- (void)dispatchForwardFusedArithmeticLayerNormalizationLayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2 forTraining:(BOOL)a3;
- (void)dispatchForwardGatherLayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2 forTraining:(BOOL)a3;
- (void)dispatchForwardLayer:(id)a0 sourceTensor:(id)a1 resultTensor:(id)a2 resultStateIsTemporary:(BOOL)a3 forTraining:(BOOL)a4;
- (void)dispatchForwardLayer:(id)a0 sourceTensor:(id)a1 secondaryTensor:(id)a2 tertiaryTensor:(id)a3 resultTensor:(id)a4 resultStateIsTemporary:(BOOL)a5 forTraining:(BOOL)a6;
- (void)dispatchForwardLossLayer:(id)a0 sourceTensor:(id)a1 labelsTensor:(id)a2 labelsTensorStride:(unsigned long long)a3 weightsTensor:(id)a4 resultTensor:(id)a5 resultStateIsTemporary:(BOOL)a6 forTraining:(BOOL)a7;
- (void)dispatchForwardMHALayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2 resultStateIsTemporary:(BOOL)a3 forTraining:(BOOL)a4;
- (void)dispatchForwardMatMulLayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2;
- (void)dispatchForwardReduceLayer:(id)a0 sourceTensor:(id)a1 resultTensor:(id)a2 reductionType:(int)a3 reduceDimensions:(id)a4 forTraining:(BOOL)a5;
- (void)dispatchForwardReshapeLayer:(id)a0 sourceTensor:(id)a1 resultTensor:(id)a2 resultStateIsTemporary:(BOOL)a3 forTraining:(BOOL)a4;
- (void)dispatchForwardScatterLayer:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2 forTraining:(BOOL)a3;
- (void)dispatchForwardSelectLayer:(id)a0 conditionTensor:(id)a1 sourceTensors:(id)a2 resultTensor:(id)a3 forTraining:(BOOL)a4;
- (void)dispatchForwardSliceLayer:(id)a0 sourceTensor:(id)a1 resultTensor:(id)a2 forTraining:(BOOL)a3;
- (void)dispatchForwardSplitLayer:(id)a0 sourceTensor:(id)a1 resultTensors:(id)a2;
- (void)dispatchGradientConcatLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensors:(id)a2;
- (void)dispatchGradientEmbeddingLayer:(id)a0 sourceGradientTensor:(id)a1;
- (void)dispatchGradientFullyConnectedLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2 resultStateIsTemporary:(BOOL)a3;
- (void)dispatchGradientLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2;
- (void)dispatchGradientLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2 secondaryResultGradientTensor:(id)a3;
- (void)dispatchGradientLossLayer:(id)a0 sourceGradientTensor:(id)a1 labelsTensor:(id)a2 labelsTensorStride:(unsigned long long)a3 weightsTensor:(id)a4 resultGradientTensor:(id)a5;
- (void)dispatchGradientMHALayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensors:(id)a2 resultStateIsTemporary:(BOOL)a3;
- (void)dispatchGradientMatMulLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensors:(id)a2;
- (void)dispatchGradientReduceLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2 reductionType:(int)a3 reduceDimensions:(id)a4;
- (void)dispatchGradientReshapeLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2;
- (void)dispatchGradientSelectLayer:(id)a0 conditionTensor:(id)a1 sourceGradientTensor:(id)a2 resultGradientTensors:(id)a3;
- (void)dispatchGradientSliceLayer:(id)a0 sourceGradientTensor:(id)a1 resultGradientTensor:(id)a2;
- (void)dispatchGradientSplitLayer:(id)a0 sourceGradientTensors:(id)a1 resultGradientTensor:(id)a2;
- (void)dispatchGradientSplitLayer:(id)a0 sourceGradientTensors:(id)a1 resultGradientTensor:(id)a2 forConcat:(BOOL)a3;
- (void)dispatchRNNForwardLayer:(id)a0 sourceTensors:(id)a1 resultTensors:(id)a2;
- (void)dispatchRNNForwardLayer:(id)a0 sourceTensors:(id)a1 resultTensors:(id)a2 resultStateIsTemporary:(BOOL)a3 forTraining:(BOOL)a4;
- (void)dispatchRNNGradientLayer:(id)a0 sourceGradientTensors:(id)a1 resultGradientTensors:(id)a2;
- (void)dispatchReadTensor:(id)a0 targetBuffer:(void *)a1 batchSize:(unsigned long long)a2;
- (id)dropoutLayerWithRate:(float)a0 seed:(unsigned long long)a1;
- (id)embeddingLayerWithDescriptor:(id)a0 weights:(id)a1 inferenceOnly:(BOOL)a2;
- (id)embeddingWeightsGradients:(id)a0 embeddingCount:(unsigned long long)a1 embeddingDimension:(unsigned long long)a2;
- (void)exportBiasGateOptimizerDataForDeviceOps:(id)a0 biasTensors:(id)a1 gateIndex:(unsigned long long)a2 optimizerDataIndex:(unsigned long long)a3;
- (id)fullyConnectedLayerWithDescriptor:(id)a0 weights:(id)a1 biasTerms:(id)a2;
- (void)fuseLayersForGraph:(id)a0 stopGradientTensorList:(id)a1 startAtLayerIndex:(unsigned long long)a2 forInference:(BOOL)a3;
- (void)fuseLayersForInferenceGraph:(id)a0 startAtLayerIndex:(unsigned long long)a1;
- (void)fuseLayersForTrainingGraph:(id)a0 stopGradientTensorList:(id)a1;
- (id)fusedBatchNormalizationAndNeuronLayerWithDescriptor:(id)a0 numOfFeatureChannels:(unsigned long long)a1 mean:(id)a2 variance:(id)a3 beta:(id)a4 gamma:(id)a5 varianceEpsilon:(float)a6 momentum:(float)a7;
- (id)fusedConvolutionAndNeuronLayerWithDescriptor:(id)a0 convolutionDescriptor:(id)a1 weights:(id)a2 biasTerms:(id)a3;
- (id)fusedConvolutionBatchNormalizationAndNeuronLayerWithDescriptor:(id)a0 mean:(id)a1 variance:(id)a2 beta:(id)a3 gamma:(id)a4 varianceEpsilon:(float)a5 momentum:(float)a6 neuronDescriptor:(id)a7 weights:(id)a8 biasTerms:(id)a9;
- (id)fusedConvolutionGroupNormalizationAndNeuronLayerWithDescriptor:(id)a0 beta:(id)a1 gamma:(id)a2 varianceEpsilon:(float)a3 momentum:(float)a4 groupCount:(unsigned long long)a5 neuronDescriptor:(id)a6 weights:(id)a7 biasTerms:(id)a8;
- (id)fusedConvolutionInstanceNormalizationAndNeuronLayerWithDescriptor:(id)a0 beta:(id)a1 gamma:(id)a2 varianceEpsilon:(float)a3 momentum:(float)a4 neuronDescriptor:(id)a5 weights:(id)a6 biasTerms:(id)a7;
- (id)fusedFullyConnectedAndNeuronLayerWithDescriptor:(id)a0 convolutionDescriptor:(id)a1 weights:(id)a2 biasTerms:(id)a3;
- (id)fusedFullyConnectedAndSoftmaxLayerWithDescriptor:(id)a0 weights:(id)a1 biasTerms:(id)a2 softmaxOp:(int)a3;
- (id)fusedFullyConnectedBatchNormalizationAndNeuronLayerWithDescriptor:(id)a0 mean:(id)a1 variance:(id)a2 beta:(id)a3 gamma:(id)a4 varianceEpsilon:(float)a5 momentum:(float)a6 neuronDescriptor:(id)a7 weights:(id)a8 biasTerms:(id)a9;
- (id)fusedGroupNormalizationAndNeuronLayerWithDescriptor:(id)a0 featureChannelCount:(unsigned long long)a1 groupCount:(unsigned long long)a2 beta:(id)a3 gamma:(id)a4 varianceEpsilon:(float)a5;
- (id)fusedInstanceNormalizationAndNeuronLayerWithDescriptor:(id)a0 numOfFeatureChannels:(unsigned long long)a1 mean:(id)a2 variance:(id)a3 beta:(id)a4 gamma:(id)a5 varianceEpsilon:(float)a6 momentum:(float)a7;
- (id)fusedLayerNormalizationAndNeuronLayerWithDescriptor:(id)a0 normalizedShape:(id)a1 beta:(id)a2 gamma:(id)a3 varianceEpsilon:(float)a4;
- (id)gatherLayerWithDimension:(unsigned long long)a0;
- (struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0[2]; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x3; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x4; })getGateDesc:(unsigned long long)a0 lstmParams:(struct { unsigned long long x0; unsigned long long x1; unsigned long long x2; unsigned long long x3; unsigned long long x4; float x5; unsigned int x6; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x7; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; } x8; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; } x9; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0[2]; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x3; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x4; } x10; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0[2]; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x3; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x4; } x11; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0[2]; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x3; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x4; } x12; struct { struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x0[2]; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x1; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x2; struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } x3; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x4; } x13; struct { unsigned int x0; float x1; float x2; int x3; int x4; int x5; int *x6; int *x7; int *x8; } x14; } *)a1;
- (id)getHostPointerIfUnifiedDeviceMemory:(id)a0;
- (id)groupNormalizationLayerWithFeatureChannelCount:(unsigned long long)a0 groupCount:(unsigned long long)a1 beta:(id)a2 gamma:(id)a3 varianceEpsilon:(float)a4;
- (void)identifySourceAndOrResultTensorsNeededByGradient:(id)a0;
- (void)incrementReadCountForGradientState:(id)a0 increment:(long long)a1;
- (void)incrementReadCountForTensorDeviceMemory:(id)a0 increment:(long long)a1;
- (id)instanceNormalizationLayerWithChannelCount:(unsigned long long)a0 mean:(id)a1 variance:(id)a2 beta:(id)a3 gamma:(id)a4 varianceEpsilon:(float)a5 momentum:(float)a6;
- (id)layerNormalizationLayerWithNormalizedShape:(id)a0 beta:(id)a1 gamma:(id)a2 varianceEpsilon:(float)a3 isFusedWithArithmeticLayer:(BOOL)a4;
- (id)lossLayerWithDescriptor:(id)a0;
- (id)lossYOLOLayerWithDescriptor:(id)a0;
- (id)lstmBiasGradient:(id)a0 mlcBiasIndex:(unsigned long long)a1;
- (id)lstmHiddenWeightGradient:(id)a0 mlcWeightIndex:(unsigned long long)a1;
- (id)lstmInputWeightGradient:(id)a0 mlcWeightIndex:(unsigned long long)a1;
- (id)lstmLayerWithDescriptor:(id)a0 inputWeights:(id)a1 hiddenWeights:(id)a2 peepholeWeights:(id)a3 biasTerms:(id)a4 gateActivations:(id)a5 outputResultActivation:(id)a6 inferenceOnly:(BOOL)a7;
- (id)matMulLayerWithDescriptor:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2 inferenceOnly:(BOOL)a3;
- (id)mhaAttnBiasGradient:(id)a0 withSize:(unsigned long long)a1 index:(unsigned long long)a2;
- (id)mhaWeightGradient:(id)a0 withSize:(unsigned long long)a1 index:(unsigned long long)a2;
- (id)multiheadAttentionLayerWithDescriptor:(id)a0 weights:(id)a1 bias:(id)a2 attnBias:(id)a3 inferenceOnly:(BOOL)a4;
- (BOOL)needToAllocateDeviceMemoryForTensor:(id)a0;
- (id)neuronLayerWithDescriptor:(id)a0;
- (void)optimizeComputationForTrainingGraph:(id)a0;
- (id)optimizerAdamWithDescriptor:(id)a0 beta1:(float)a1 beta2:(float)a2 epsilon:(float)a3 amsgrad:(BOOL)a4 timeStep:(unsigned long long)a5 isAdamW:(BOOL)a6;
- (id)optimizerRMSPropWithDescriptor:(id)a0 momentumScale:(float)a1 alpha:(float)a2 epsilon:(float)a3 centered:(BOOL)a4;
- (id)optimizerSGDWithDescriptor:(id)a0 momentunScale:(float)a1 useNesterovMomentum:(BOOL)a2;
- (void)optimizerStepForSingleParameterLSTM:(id)a0 tensorParameters:(id)a1 parameterForGateDesc:(struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } *)a2 gradientsForGateDesc:(struct { unsigned int x0; unsigned int x1; unsigned long long x2[8]; unsigned long long x3[8]; void *x4; unsigned int x5; void *x6; unsigned int x7; float x8; float x9; } *)a3 parameterMomentumDescData:(id)a4 gateIndex:(unsigned long long)a5 deviceOptimizers:(id)a6 isStackedInputWeight:(BOOL)a7;
- (id)paddingLayerWithPaddingType:(int)a0 paddingLeft:(unsigned long long)a1 paddingRight:(unsigned long long)a2 paddingTop:(unsigned long long)a3 paddingBottom:(unsigned long long)a4 constantValue:(float)a5 dataType:(int)a6;
- (id)poolingLayerWithDescriptor:(id)a0 paddingSizes:(unsigned long long *)a1;
- (id)readFromDevice:(id)a0 deviceIndex:(unsigned long long)a1 allocateData:(BOOL)a2 batchSize:(unsigned long long)a3;
- (id)readTensor:(id)a0;
- (id)readTensor:(id)a0 fromDeviceIndex:(unsigned long long)a1;
- (id)readTensor:(id)a0 fromDeviceIndex:(unsigned long long)a1 batchSize:(unsigned long long)a2;
- (void)readTensor:(id)a0 fromDeviceIndex:(unsigned long long)a1 targetBuffer:(void *)a2 batchSize:(unsigned long long)a3;
- (void)readTensor:(id)a0 fromDeviceIndex:(unsigned long long)a1 targetBuffer:(void *)a2 batchSize:(unsigned long long)a3 reduceOverBatch:(BOOL)a4;
- (void)readTensor:(id)a0 targetBuffer:(void *)a1;
- (id)reductionLayerWithReduceType:(int)a0 dimensions:(id)a1 sourceShapeCount:(unsigned long long)a2;
- (void)reloadParameterDataFromHostToDeviceMemoryForTensor:(id)a0;
- (id)reshapeLayerWithShape:(id)a0;
- (void)restoreRNNParamsWithDeviceOps:(id)a0 optimizer:(id)a1 inputWeightsParameter:(id)a2 hiddenWeightsParameter:(id)a3 biasesParameter:(id)a4 arrayOfParams:(id)a5;
- (void)saveOrRestoreLSTMBiasesAndAccumulators:(id)a0 auxParameter:(id)a1 auxAccumulators:(id)a2 mlcIndex:(unsigned long long)a3 auxIndex:(unsigned long long)a4 saveParameter:(BOOL)a5;
- (id)scatterLayerWithDimension:(unsigned long long)a0 reduceType:(int)a1;
- (id)selectLayer;
- (void)setConvolutionGradientComputeWeightsAndBiasOnly:(id)a0;
- (BOOL)setConvolutionLayerOptimizerDataForDeviceOps:(id)a0 weights:(id)a1 bias:(id)a2;
- (void)setDeviceMemoryForTensor:(id)a0 data:(id)a1;
- (void)setFullyConnectedGradientComputeWeightsAndBiasOnly:(id)a0;
- (BOOL)setLSTMLayerOptimizerDataForDeviceOps:(id)a0 inputWeights:(id)a1 hiddenWeights:(id)a2 biasTerms:(id)a3;
- (BOOL)setMHALayerOptimizerDataForDeviceOps:(id)a0 optimizerDataForWeights:(id)a1 optimizerDataForBias:(id)a2;
- (BOOL)setNormalizationLayerOptimizerDataForDeviceOps:(id)a0 beta:(id)a1 gamma:(id)a2;
- (void)setOptimizerLearningRate:(id)a0 learningRate:(float)a1;
- (void)setOptimizerTimeStep:(id)a0 timeStep:(unsigned long long)a1;
- (BOOL)shareDeviceMemoryWithResultTensor:(id)a0 sourceTensor:(id)a1;
- (void)shareResultTensor:(id)a0 sourceTensor:(id)a1 deviceOps:(id)a2;
- (void)signalAllDevicesExcludingDevice:(unsigned long long)a0 eventValue:(unsigned long long)a1;
- (id)sliceLayerWithbegin:(id)a0 end:(id)a1 stride:(id)a2 inferenceOnly:(BOOL)a3;
- (id)softmaxLayerWithOperation:(int)a0 dimension:(unsigned long long)a1 sourceShapeCount:(unsigned long long)a2;
- (id)splitLayerWithDimension:(unsigned long long)a0;
- (void)sumSharedGradientsForConvolutionLayerTensorParameter:(id)a0 layerIndexForSummedGradients:(unsigned long long)a1;
- (void)sumSharedGradientsForNormalizationLayerTensorParameter:(id)a0 layerIndexForSummedGradients:(unsigned long long)a1 isBetaTensor:(BOOL)a2;
- (void)synchronizeOptimizerUpdatesForTensor:(id)a0;
- (BOOL)synchronizeTensor:(id)a0;
- (void)synchronizeUpdatesForLayer:(id)a0;
- (BOOL)transferTensor:(id)a0 fromDevice:(id)a1;
- (id)transposeLayerWithShape:(id)a0;
- (void)updateAllParametersWithOptimizer:(id)a0 arrayOfParameters:(id)a1;
- (void)updateBatchNormalizationLayer:(id)a0 optimizer:(id)a1 betaParameter:(id)a2 gammaParameter:(id)a3 meanTensor:(id)a4 varianceTensor:(id)a5 arrayOfParams:(id)a6;
- (void)updateConvolutionLayer:(id)a0 optimizer:(id)a1 weightsParameter:(id)a2 biasesParameter:(id)a3 arrayOfParams:(id)a4;
- (BOOL)updateDeviceMemoryForTensor:(id)a0;
- (void)updateEmbeddingLayer:(id)a0 weightsParameter:(id)a1 optimizer:(id)a2 arrayOfParams:(id)a3;
- (void)updateFullyConnectedLayer:(id)a0 optimizer:(id)a1 weightsParameter:(id)a2 biasesParameter:(id)a3 arrayOfParams:(id)a4;
- (void)updateGradientDescriptorsForDeviceOps:(id)a0 sourceTensors:(id)a1 resultTensor:(id)a2;
- (void)updateGroupNormalizationLayer:(id)a0 optimizer:(id)a1 betaParameter:(id)a2 gammaParameter:(id)a3 arrayOfParams:(id)a4;
- (void)updateInstanceNormalizationLayer:(id)a0 optimizer:(id)a1 betaParameter:(id)a2 gammaParameter:(id)a3 arrayOfParams:(id)a4;
- (void)updateLayerNormalizationLayer:(id)a0 optimizer:(id)a1 betaParameter:(id)a2 gammaParameter:(id)a3 arrayOfParams:(id)a4;
- (void)updateMultiheadAttentionLayer:(id)a0 optimizer:(id)a1 weightsParameter:(id)a2 biasesParameter:(id)a3 arrayOfParams:(id)a4;
- (void)updateRNNLayer:(id)a0 optimizer:(id)a1 inputWeightsParameter:(id)a2 hiddenWeightsParameter:(id)a3 biasesParameter:(id)a4 arrayOfParams:(id)a5;
- (void)updateTensorParameter:(id)a0 optimizer:(id)a1 gradient:(id)a2 arrayOfParams:(id)a3;
- (id)upsampleLayerWithScaleFactorX:(float)a0 scaleFactorY:(float)a1 sampleMode:(int)a2 alignCorners:(BOOL)a3;
- (void)waitForAllDevicesExcludingDevice:(unsigned long long)a0 eventValue:(unsigned long long)a1;
- (void)waitForOthers;
- (id)weightsGradients:(id)a0;
- (void)writeToAllDevices:(id)a0 allocateData:(BOOL)a1 batchSize:(unsigned long long)a2;
- (void)writeToDevice:(id)a0 batchSize:(unsigned long long)a1;
- (void)writeToDevice:(id)a0 sourceBuffer:(void *)a1 batchSize:(unsigned long long)a2;
@end