@class NSString, NSArray;

@interface MLCDeviceANE : 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 (readonly) unsigned long long hash;
@property (readonly) Class superclass;
@property (readonly, copy) NSString *description;
@property (readonly, copy) NSString *debugDescription;

+ (BOOL)hasANE;

- (void).cxx_destruct;
- (id)initWithType:(int)a0;
- (id)copyWithZone:(struct _NSZone { } *)a0;
- (unsigned long long)numDevices;
- (id)ANERequestWithModelAttributes:(id)a0 procedureName:(id)a1 tensorLabelToIOSurfaceMap:(id)a2;
- (void)dispatchForwardEmbeddingLayer:(id)a0 weight:(id)a1 sourceTensor:(id)a2 resultTensor:(id)a3;
- (id)gramMatrixLayerWithScaleFactor:(float)a0;
- (void)signalNextEvent;
- (unsigned long long)numberOfDevices;
- (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;
- (BOOL)allocateDeviceMemoryForSourceAndResultTensorsOfLayer:(id)a0 tensorLabelToIOSurfaceMap:(id *)a1;
- (void)allocateDeviceMemoryForTensor:(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;
- (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)compileOptimizerDeviceOps:(id)a0;
- (id)concatLayerWithConcatDimension:(unsigned long long)a0 sourceShapeCount:(unsigned long long)a1;
- (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;
- (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 interleave:(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;
- (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)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)fullyConnectedLayerWithDescriptor:(id)a0 weights:(id)a1 biasTerms:(id)a2;
- (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)fusedFullyConnectedAndNeuronLayerWithDescriptor:(id)a0 convolutionDescriptor:(id)a1 weights:(id)a2 biasTerms:(id)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)getHostPointerIfUnifiedDeviceMemory:(id)a0;
- (id)groupNormalizationLayerWithFeatureChannelCount:(unsigned long long)a0 groupCount:(unsigned long long)a1 beta:(id)a2 gamma:(id)a3 varianceEpsilon:(float)a4;
- (id)imageReshapeWithWidth:(unsigned long long)a0 height:(unsigned long long)a1 alignCorners:(BOOL)a2;
- (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)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)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:(double)a1 beta2:(double)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;
- (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)partitionInferenceGraph:(id)a0 startAtLayerIndex:(unsigned long long)a1 aneDevice:(id)a2 secondaryDevice:(id)a3;
- (id)partitionInferenceGraph:(id)a0 startAtLayerIndex:(unsigned long long)a1 aneDevice:(id)a2 secondaryDevice:(id)a3 configurationJSON:(id)a4;
- (id)poolingLayerWithDescriptor:(id)a0 paddingSizes:(unsigned long long *)a1;
- (BOOL)postProcessCompiledGraph:(id)a0 compilerOptions:(unsigned long long)a1;
- (BOOL)procedureInformationWithModelAttributes:(id)a0 procedureName:(id)a1 procedureID:(unsigned long long *)a2 procedureInputSymbols:(id *)a3 procedureInputSymbolIndices:(id *)a4 procedureOutputSymbols:(id *)a5 procedureOutputSymbolIndices:(id *)a6;
- (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;
- (void)resetLayer:(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)saveGraphPartitioning:(id)a0 toFile:(id)a1;
- (id)selectLayer;
- (BOOL)setConvolutionLayerOptimizerDataForDeviceOps:(id)a0 weights:(id)a1 bias:(id)a2;
- (void)setDeviceMemoryForTensor:(id)a0 data:(id)a1;
- (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)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;
- (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)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;
- (BOOL)updateTensorsForFusedLayers:(id)a0 ofInferenceGraph:(id)a1;
- (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;
- (void)writeToAllDevices:(id)a0 allocateData:(BOOL)a1 batchSize:(unsigned long long)a2;
- (void)writeToDevice:(id)a0 toDeviceIndex:(unsigned long long)a1 batchSize:(unsigned long long)a2;

@end