diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-07 11:39:37 +0100 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-07 11:40:21 +0100 |
commit | 3083f7ee68ce08147db08fca2474e5f4712fc8d7 (patch) | |
tree | c52e668c01a6a1041c08190e52a15944fd65b453 /tests/test_nn_tensorflow_tflite_metrics.py | |
parent | bb7fb49484bb3687041061b2fdbbfae3959be54b (diff) | |
download | mlia-3083f7ee68ce08147db08fca2474e5f4712fc8d7.tar.gz |
MLIA-607 Update documentation and comments
Use "TensorFlow Lite" instead of "TFLite" in
documentation and comments
Change-Id: Ie4450d72fb2e5261d152d72ab8bd94c3da914c46
Diffstat (limited to 'tests/test_nn_tensorflow_tflite_metrics.py')
-rw-r--r-- | tests/test_nn_tensorflow_tflite_metrics.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_nn_tensorflow_tflite_metrics.py b/tests/test_nn_tensorflow_tflite_metrics.py index ca4ab55..0e4c79c 100644 --- a/tests/test_nn_tensorflow_tflite_metrics.py +++ b/tests/test_nn_tensorflow_tflite_metrics.py @@ -53,7 +53,7 @@ def _sparse_binary_keras_model() -> tf.keras.Model: @pytest.fixture(scope="class", name="tflite_file") def fixture_tflite_file() -> Generator: - """Generate temporary TFLite file for tests.""" + """Generate temporary TensorFlow Lite file for tests.""" converter = tf.lite.TFLiteConverter.from_keras_model(_sparse_binary_keras_model()) tflite_model = converter.convert() with tempfile.TemporaryDirectory() as tmp_dir: @@ -64,7 +64,7 @@ def fixture_tflite_file() -> Generator: @pytest.fixture(scope="function", name="metrics") def fixture_metrics(tflite_file: str) -> TFLiteMetrics: - """Generate metrics file for a given TFLite model.""" + """Generate metrics file for a given TensorFlow Lite model.""" return TFLiteMetrics(tflite_file) @@ -74,7 +74,7 @@ class TestTFLiteMetrics: @staticmethod def test_sparsity(metrics: TFLiteMetrics) -> None: """Test sparsity.""" - # Create new instance with a sample TFLite file + # Create new instance with a sample TensorFlow Lite file # Check sparsity calculation sparsity_per_layer = metrics.sparsity_per_layer() for name, sparsity in sparsity_per_layer.items(): |