Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
'LeakyReLU' : ParametrizedActivation,
'ThresholdedReLU' : ParametrizedActivation,
'ELU' : ParametrizedActivation,
'PReLU' : PReLU,
'Dense' : Dense,
'BinaryDense' : Dense,
'TernaryDense' : Dense,
'Conv1D' : Conv1D,
'Conv2D' : Conv2D,
'BatchNormalization' : BatchNormalization,
'MaxPooling1D' : Pooling1D,
'AveragePooling1D' : Pooling1D,
'MaxPooling2D' : Pooling2D,
'AveragePooling2D' : Pooling2D,
'Merge' : Merge,
'Concatenate' : Concatenate,
}
def register_layer(name, clazz):
global layer_map
layer_map[name] = clazz