@interface _MLCGPUBatchNormalization : _MLCGPULayer

+ (id)layerWithDevice:(id)a0 fusedWithNeuronDescriptor:(id)a1 numOfFeatureChannels:(unsigned long long)a2 mean:(id)a3 variance:(id)a4 beta:(id)a5 gamma:(id)a6 varianceEpsilon:(float)a7 momentum:(float)a8;
+ (id)layerWithDevice:(id)a0 numOfFeatureChannels:(unsigned long long)a1 mean:(id)a2 variance:(id)a3 beta:(id)a4 gamma:(id)a5 varianceEpsilon:(float)a6 momentum:(float)a7;
+ (BOOL)setOptimizerDataForDevice:(id)a0 deviceOps:(id)a1 beta:(id)a2 gamma:(id)a3;

- (id)initWithDevice:(id)a0 fusedWithNeuronDescriptor:(id)a1 numOfFeatureChannels:(unsigned long long)a2 mean:(id)a3 variance:(id)a4 beta:(id)a5 gamma:(id)a6 varianceEpsilon:(float)a7 momentum:(float)a8;
- (id)initWithDevice:(id)a0 numOfFeatureChannels:(unsigned long long)a1 mean:(id)a2 variance:(id)a3 beta:(id)a4 gamma:(id)a5 varianceEpsilon:(float)a6 momentum:(float)a7;

@end