@protocol MTLBuffer;
@interface MPSCNNDropoutGradientState : MPSNNGradientState {
struct { unsigned long long width; unsigned long long height; unsigned long long depth; } _maskStrideInPixels;
float _keepProbability;
unsigned long long _resourcePixelFormat;
struct { unsigned long long width; unsigned long long height; unsigned long long depth; } _resourceSize;
unsigned long long _commonBufferOffsetBytes;
id<MTLBuffer> _commonMaskBuffer;
}
- (id)init;
- (void)dealloc;
- (void)synchronizeOnCommandBuffer:(id)a0;
- (id)maskData;
@end