diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/prelu/__init__.py')
| -rwxr-xr-x | res/TensorFlowPythonExamples/examples/prelu/__init__.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/prelu/__init__.py b/res/TensorFlowPythonExamples/examples/prelu/__init__.py new file mode 100755 index 000000000..2ab030265 --- /dev/null +++ b/res/TensorFlowPythonExamples/examples/prelu/__init__.py @@ -0,0 +1,7 @@ +import tensorflow as tf +from tensorflow.compat.v1.keras import layers + +model = tf.compat.v1.keras.Sequential() +model.add(layers.PReLU()) +# TODO Find a way to freeze Keras model for inference +model.build((1, 1)) |