diff options
Diffstat (limited to 'src/mlia/nn/rewrite/library/fc_layer.py')
| -rw-r--r-- | src/mlia/nn/rewrite/library/fc_layer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/mlia/nn/rewrite/library/fc_layer.py b/src/mlia/nn/rewrite/library/fc_layer.py index 041ce85..cb98cb9 100644 --- a/src/mlia/nn/rewrite/library/fc_layer.py +++ b/src/mlia/nn/rewrite/library/fc_layer.py @@ -6,7 +6,7 @@ from typing import Any from keras.api._v2 import keras # Temporary workaround for now: MLIA-1107 -def get_keras_model(input_shape: Any, output_shape: Any) -> keras.Model: +def fc_rewrite(input_shape: Any, output_shape: Any) -> keras.Model: """Generate TensorFlow Lite model for rewrite.""" model = keras.Sequential( ( |