1 #import <MetalPerformanceShaders/MetalPerformanceShaders.h> 2 3 @interface MPSCNNNeuronOp : NSObject 4 5 + (MPSCNNNeuronHardSigmoid*)hardSigmoid API_AVAILABLE(ios(11.0), macos(10.13)); 6 + (MPSCNNNeuronReLU*)relu; 7 + (MPSCNNNeuronSigmoid*)sigmoid; 8 + (MPSCNNNeuronTanH*)tanh; 9 10 @end 11 12 API_AVAILABLE(ios(11.3), macos(10.13), macCatalyst(13.0)) 13 @interface MPSCNNNeuronOpDescriptor : NSObject 14 15 + (MPSNNNeuronDescriptor*)hardSigmoidDescriptor; 16 + (MPSNNNeuronDescriptor*)reluDescriptor; 17 + (MPSNNNeuronDescriptor*)sigmoidDescriptor; 18 + (MPSNNNeuronDescriptor*)tanhDescriptor; 19 20 @end 21