@class NSData, MPSImage;
@interface MPSCNNLossLabels : MPSState {
struct { unsigned long long width; unsigned long long height; unsigned long long depth; } _lossImageSize;
BOOL _isScalarLoss;
NSData *_userData;
unsigned long long _userDataLayout;
unsigned long long _numFeatureChannels_labels;
unsigned long long _numFeatureChannels_loss;
MPSImage *_userLabelsImage;
MPSImage *_userWeightsImage;
BOOL _hasStateWeights;
BOOL _computeNonZeroWeights;
}
- (id)init;
- (void)dealloc;
- (id)initWithDevice:(id)a0 lossImageSize:(struct { unsigned long long x0; unsigned long long x1; unsigned long long x2; })a1 labelsImage:(id)a2 weightsImage:(id)a3;
- (id)lossImage;
- (id)initWithDevice:(id)a0 lossImageSize:(struct { unsigned long long x0; unsigned long long x1; unsigned long long x2; })a1 labelsDescriptor:(id)a2 weightsDescriptor:(id)a3;
- (id)labelsImage;
- (id)weightsImage;
- (id)initWithDevice:(id)a0 labelsDescriptor:(id)a1;
@end