System/Library/Frameworks/MLCompute.framework

_MLCCPUMHAttention.h

Download _MLCCPUMHAttention.h

1. NSObject
2. >
3. _MLCCPULayer

@interface _MLCCPUMHAttention : _MLCCPULayer

+ (id)layerWithDevice:(id)a0 descriptor:(id)a1 weights:(id)a2 bias:(id)a3 attnBias:(id)a4 inferenceOnly:(BOOL)a5;
+ (BOOL)setOptimizerDataForDevice:(id)a0 deviceOps:(id)a1 dataForWeights:(id)a2 dataForBias:(id)a3;
+ (BOOL)compileWithDevice:(id)a0 deviceOps:(id)a1 sourceTensors:(id)a2 resultTensor:(id)a3;

- (id)initWithDevice:(id)a0 descriptor:(id)a1 weights:(id)a2 bias:(id)a3 attnBias:(id)a4 inferenceOnly:(BOOL)a5;

@end