From 3083f7ee68ce08147db08fca2474e5f4712fc8d7 Mon Sep 17 00:00:00 2001 From: Dmitrii Agibov Date: Fri, 7 Oct 2022 11:39:37 +0100 Subject: MLIA-607 Update documentation and comments Use "TensorFlow Lite" instead of "TFLite" in documentation and comments Change-Id: Ie4450d72fb2e5261d152d72ab8bd94c3da914c46 --- tests/test_nn_tensorflow_tflite_metrics.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tests/test_nn_tensorflow_tflite_metrics.py') 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(): -- cgit v1.2.1