@class TCMPSStyleTransferTransformerNetwork, TCMPSStyleTransferPreProcessing, TCMPSVgg16Network, MPSNNImageNode, MPSNNGraph;
@protocol MTLCommandQueue, MTLDevice;

@interface TCMPSStyleTransfer : NSObject

@property (retain, nonatomic) TCMPSStyleTransferTransformerNetwork *model;
@property (retain, nonatomic) TCMPSStyleTransferPreProcessing *contentPreProcess;
@property (retain, nonatomic) TCMPSStyleTransferPreProcessing *stylePreProcessLoss;
@property (retain, nonatomic) TCMPSStyleTransferPreProcessing *contentPreProcessLoss;
@property (retain, nonatomic) TCMPSVgg16Network *contentVgg;
@property (retain, nonatomic) TCMPSVgg16Network *styleVggLoss;
@property (retain, nonatomic) TCMPSVgg16Network *contentVggLoss;
@property (retain, nonatomic) MPSNNImageNode *contentNode;
@property (retain, nonatomic) MPSNNImageNode *contentScaleNode;
@property (retain, nonatomic) MPSNNImageNode *contenMeanNode;
@property (retain, nonatomic) MPSNNImageNode *styleNode;
@property (retain, nonatomic) MPSNNImageNode *styleScaleNode;
@property (retain, nonatomic) MPSNNImageNode *styleMeanNode;
@property (retain, nonatomic) MPSNNGraph *trainingGraph;
@property (retain, nonatomic) MPSNNGraph *inferenceGraph;
@property (retain, nonatomic) id<MTLDevice> dev;
@property (retain, nonatomic) id<MTLCommandQueue> commandQueue;
@property (nonatomic) unsigned long long batchSize;
@property (nonatomic) float contentLossMultiplier;
@property (nonatomic) float styleLossMultiplier;
@property (nonatomic) float totalLossMultiplier;
@property (nonatomic) BOOL updateAllParams;
@property (nonatomic) unsigned long long imgWidth;
@property (nonatomic) unsigned long long imgHeight;

+ (id)defineTransformerDescriptor:(unsigned long long)a0 tuneAllWeights:(BOOL)a1;
+ (id)defineVGG16Descriptor:(unsigned long long)a0;
+ (void)populateMean:(id)a0;
+ (void)populateMeanCWH:(id)a0 height:(unsigned long long)a1 width:(unsigned long long)a2;
+ (void)populateMultiplication:(id)a0;
+ (void)populateVarianceCWH:(id)a0 height:(unsigned long long)a1 width:(unsigned long long)a2;
+ (id)topLeftCropImage:(id)a0 inputHeight:(unsigned long long)a1 inputWidth:(unsigned long long)a2 outputHeight:(unsigned long long)a3 outputWidth:(unsigned long long)a4 numChannels:(unsigned long long)a5;

- (void)checkpoint;
- (void).cxx_destruct;
- (id)predict:(id)a0;
- (void)setLearningRate:(float)a0;
- (id)train:(id)a0;
- (id)exportWeights;
- (id)initWithDev:(id)a0 commandQueue:(id)a1 weights:(id)a2 numStyles:(unsigned long long)a3;

@end