aboutsummaryrefslogtreecommitdiff
path: root/tests/test_nn_tensorflow_tflite_metrics.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-07 11:39:37 +0100
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-07 11:40:21 +0100
commit3083f7ee68ce08147db08fca2474e5f4712fc8d7 (patch)
treec52e668c01a6a1041c08190e52a15944fd65b453 /tests/test_nn_tensorflow_tflite_metrics.py
parentbb7fb49484bb3687041061b2fdbbfae3959be54b (diff)
downloadmlia-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.py6
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():