Path	Lines of Code
projects/scale_hyperprior_lightning/tests/test_scale_hyperprior_lightning.py	25
projects/variational_image_compression/test/__init__.py	1
projects/variational_image_compression/test/test_factorized_prior_autoencoder.py	102
projects/variational_image_compression/test/test_mean_scale_hyperprior_autoencoder.py	103
projects/variational_image_compression/test/test_scale_hyperprior_autoencoder.py	103
tests/data/__init__.py	1
tests/data/test_clic_2020_image.py	22
tests/data/test_clic_2020_video.py	31
tests/data/test_kodak.py	31
tests/data/test_vimeo_90k_septuplet.py	71
tests/distributions/__init__.py	1
tests/distributions/test_noisy_normal.py	4
tests/distributions/test_uniform_noise.py	91
tests/functional/__init__.py	1
tests/functional/test_count_flops.py	104
tests/functional/test_dense_image_warp.py	42
tests/functional/test_estimate_tails.py	10
tests/functional/test_hsv_to_rgb.py	19
tests/functional/test_information_content.py	30
tests/functional/test_log_cdf.py	22
tests/functional/test_log_expm1.py	13
tests/functional/test_log_ndtr.py	14
tests/functional/test_log_survival_function.py	22
tests/functional/test_lower_bound.py	22
tests/functional/test_lower_tail.py	19
tests/functional/test_lpips.py	26
tests/functional/test_multiscale_structural_similarity.py	58
tests/functional/test_ndtr.py	16
tests/functional/test_optical_flow_to_rgb.py	59
tests/functional/test_pmf_to_quantized_cdf.py	13
tests/functional/test_quantization_offset.py	6
tests/functional/test_soft_round.py	16
tests/functional/test_soft_round_conditional_mean.py	11
tests/functional/test_soft_round_inverse.py	21
tests/functional/test_survival_function.py	16
tests/functional/test_upper_tail.py	19
tests/layers/__init__.py	1
tests/layers/test_analysis_transformation_2d.py	7
tests/layers/test_continuous_entropy.py	168
tests/layers/test_generalized_divisive_normalization.py	29
tests/layers/test_hyper_analysis_transformation_2d.py	7
tests/layers/test_hyper_synthesis_transformation_2d.py	8
tests/layers/test_non_negative_parameterization.py	31
tests/layers/test_rate_mse_distortion_loss.py	15
tests/layers/test_synthesis_transformation_2d.py	7
tests/models/__init__.py	1
tests/models/_hific/test_hific_discriminator.py	10
tests/models/_hific/test_hific_encoder.py	7
tests/models/_hific/test_hific_generator.py	8
tests/models/test_factorized_prior_autoencoder.py	25
tests/models/test_mean_scale_hyperprior_autoencoder.py	19
tests/models/test_prior_autoencoder.py	40
tests/models/test_scale_hyperprior_autoencoder.py	27
tests/test_entropy_coders.py	393
tests/test_layers_gdn.py	61
tests/test_models.py	91
tests/test_scale_hyperprior.py	26
tests/utils.py	19