System/Library/Frameworks/MLCompute.framework

MLCMultiGPUDeviceOps.h

Download MLCMultiGPUDeviceOps.h

1. NSObject
2. >
3. MLCDeviceOps
4. >
5. MLCGPUDeviceOps

@class NSArray, NSMutableArray;

@interface MLCMultiGPUDeviceOps : MLCGPUDeviceOps

@property (nonatomic) int multiGPUReduction;
@property (retain, nonatomic) NSArray *localGradientMatrices;
@property (retain, nonatomic) NSArray *remoteGradientBuffers;
@property (retain, nonatomic) NSArray *receivedGradientMatrices;
@property (retain, nonatomic) NSArray *matrixSumKernels;
@property (retain, nonatomic) NSMutableArray *allReducedGradientMatrices;
@property (retain, nonatomic) NSMutableArray *rnnTrainableWeightGradients;

+ (id)multiGPUDeviceOpsWithGPUDeviceOps:(id)a0;

- (void).cxx_destruct;
- (id)initWithGPUDevicOps:(id)a0;

@end