public static bool IsActivationFusable(Layer.Activation activationType) { var fusedActivationType = (Layer.FusedActivation)activationType; switch (fusedActivationType) { case Layer.FusedActivation.None: case Layer.FusedActivation.Relu: case Layer.FusedActivation.Tanh: case Layer.FusedActivation.Softplus: case Layer.FusedActivation.Sigmoid: case Layer.FusedActivation.Relu6: case Layer.FusedActivation.Swish: case Layer.FusedActivation.Neg: case Layer.FusedActivation.Sqrt: case Layer.FusedActivation.Exp: case Layer.FusedActivation.Log: case Layer.FusedActivation.Acos: case Layer.FusedActivation.Acosh: case Layer.FusedActivation.Asin: case Layer.FusedActivation.Asinh: case Layer.FusedActivation.Atan: case Layer.FusedActivation.Atanh: case Layer.FusedActivation.Cos: case Layer.FusedActivation.Cosh: case Layer.FusedActivation.Sin: case Layer.FusedActivation.Sinh: case Layer.FusedActivation.Tan: case Layer.FusedActivation.Erf: return(true); default: return(false); } }
private Layer Activation(Layer.Activation activation, string name, object input) { Layer layer = new Layer(name, activation); layer.inputs = new [] { ResolveInput(input) }; m_Model.layers.Add(layer); return(layer); }
public static bool IsActivationFusable(Layer.Activation activationType) { var fusedActivationType = (Layer.FusedActivation)activationType; return(Enum.IsDefined(typeof(Layer.FusedActivation), fusedActivationType)); }