__init__.py « prelu « examples « TensorFlowPythonExamples « res - platform/core/ml/nnfw - Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

index : platform/core/ml/nnfw

Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

Sangmin Seo <sangmin7.seo@samsung.com>, Sung-Jae Lee <sjlee@mail.com>

summary refs log tree commit diff

path: root/res/TensorFlowPythonExamples/examples/prelu/__init__.py

blob: 2ab0302656cac9a50fd9975db913a3af61d63282 (plain)

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))