System/Library/Frameworks/MetalPerformanceShadersGraph.framework

MPSGraphPruningGradientOp.h

Download MPSGraphPruningGradientOp.h

1. NSObject
2. >
3. MPSGraphObject
4. >
5. MPSGraphOperation

@class NSArray, MPSGraphPruningDescriptor;

@interface MPSGraphPruningGradientOp : MPSGraphOperation {
    MPSGraphPruningDescriptor *_pruningDesc;
    NSArray *_shape;
}

- (void).cxx_destruct;
- (id)initWithGraph:(id)a0 inputTensors:(id)a1 controlDependencies:(id)a2 shape:(id)a3 descriptor:(id)a4 name:(id)a5;
- (void *)makeMLIROpWithBuilder:(void *)a0 symbolTable:(void *)a1 inputValues:(void *)a2 opInitialization:(BOOL)a3 name:(id)a4;

@end