Path Lines of Code ci/kokoro/gcp_ubuntu/common.cfg 7 ci/kokoro/gcp_ubuntu/continuous.cfg 1 ci/kokoro/gcp_ubuntu/presubmit.cfg 1 setup.py 62 tensorflow_model_optimization/__init__.py 46 tensorflow_model_optimization/g3doc/_book.yaml 65 tensorflow_model_optimization/g3doc/_index.yaml 100 tensorflow_model_optimization/g3doc/tools/build_docs.py 44 tensorflow_model_optimization/python/__init__.py 1 tensorflow_model_optimization/python/core/__init__.py 1 tensorflow_model_optimization/python/core/api/__init__.py 4 tensorflow_model_optimization/python/core/api/clustering/__init__.py 1 tensorflow_model_optimization/python/core/api/clustering/keras/__init__.py 9 tensorflow_model_optimization/python/core/api/clustering/keras/experimental/__init__.py 1 tensorflow_model_optimization/python/core/api/experimental/__init__.py 1 tensorflow_model_optimization/python/core/api/experimental/combine/__init__.py 6 tensorflow_model_optimization/python/core/api/quantization/__init__.py 1 tensorflow_model_optimization/python/core/api/quantization/keras/__init__.py 16 tensorflow_model_optimization/python/core/api/quantization/keras/collaborative_optimizations/__init__.py 2 tensorflow_model_optimization/python/core/api/quantization/keras/default_8bit/__init__.py 4 tensorflow_model_optimization/python/core/api/quantization/keras/default_8bit/default_8bit_transforms/__init__.py 16 tensorflow_model_optimization/python/core/api/quantization/keras/experimental/__init__.py 1 tensorflow_model_optimization/python/core/api/quantization/keras/experimental/default_n_bit/__init__.py 4 tensorflow_model_optimization/python/core/api/quantization/keras/experimental/default_n_bit/default_n_bit_transforms/__init__.py 16 tensorflow_model_optimization/python/core/api/quantization/keras/graph_transformations/__init__.py 2 tensorflow_model_optimization/python/core/api/quantization/keras/graph_transformations/model_transformer/__init__.py 1 tensorflow_model_optimization/python/core/api/quantization/keras/graph_transformations/transforms/__init__.py 3 tensorflow_model_optimization/python/core/api/quantization/keras/quantizers/__init__.py 4 tensorflow_model_optimization/python/core/api/sparsity/__init__.py 1 tensorflow_model_optimization/python/core/api/sparsity/keras/__init__.py 11 tensorflow_model_optimization/python/core/clustering/__init__.py 1 tensorflow_model_optimization/python/core/clustering/keras/__init__.py 1 tensorflow_model_optimization/python/core/clustering/keras/cluster.py 97 tensorflow_model_optimization/python/core/clustering/keras/cluster_config.py 9 tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py 296 tensorflow_model_optimization/python/core/clustering/keras/clusterable_layer.py 9 tensorflow_model_optimization/python/core/clustering/keras/clustering_algorithm.py 112 tensorflow_model_optimization/python/core/clustering/keras/clustering_callbacks.py 47 tensorflow_model_optimization/python/core/clustering/keras/clustering_centroids.py 204 tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py 124 tensorflow_model_optimization/python/core/clustering/keras/experimental/__init__.py 1 tensorflow_model_optimization/python/core/clustering/keras/experimental/cluster.py 14 tensorflow_model_optimization/python/core/common/keras/compression/algorithm.py 56 tensorflow_model_optimization/python/core/common/keras/compression/algorithms/bias_only.py 43 tensorflow_model_optimization/python/core/common/keras/compression/algorithms/different_training_and_inference.py 55 tensorflow_model_optimization/python/core/common/keras/compression/algorithms/periodical_update_and_scheduling.py 91 tensorflow_model_optimization/python/core/common/keras/compression/algorithms/same_training_and_inference.py 53 tensorflow_model_optimization/python/core/common/keras/compression/algorithms/weight_clustering.py 77 tensorflow_model_optimization/python/core/common/keras/compression/internal/optimize.py 210 tensorflow_model_optimization/python/core/common/keras/compression/schedules.py 47 tensorflow_model_optimization/python/core/internal/__init__.py 1 tensorflow_model_optimization/python/core/internal/tensor_encoding/__init__.py 8 tensorflow_model_optimization/python/core/internal/tensor_encoding/core/__init__.py 12 tensorflow_model_optimization/python/core/internal/tensor_encoding/core/core_encoder.py 251 tensorflow_model_optimization/python/core/internal/tensor_encoding/core/encoding_stage.py 234 tensorflow_model_optimization/python/core/internal/tensor_encoding/core/gather_encoder.py 235 tensorflow_model_optimization/python/core/internal/tensor_encoding/core/simple_encoder.py 88 tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/__init__.py 8 tensorflow_model_optimization/python/core/internal/tensor_encoding/encoders/common_encoders.py 39 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/__init__.py 9 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/__init__.py 11 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/clipping.py 79 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/kashin.py 147 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/misc.py 94 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/research/quantization.py 208 tensorflow_model_optimization/python/core/internal/tensor_encoding/stages/stages_impl.py 274 tensorflow_model_optimization/python/core/internal/tensor_encoding/testing/__init__.py 17 tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/__init__.py 15 tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/py_utils.py 78 tensorflow_model_optimization/python/core/internal/tensor_encoding/utils/tf_utils.py 312 tensorflow_model_optimization/python/core/keras/__init__.py 1 tensorflow_model_optimization/python/core/keras/compat.py 14 tensorflow_model_optimization/python/core/keras/metrics.py 50 tensorflow_model_optimization/python/core/keras/testing/__init__.py 1 tensorflow_model_optimization/python/core/keras/utils.py 12 tensorflow_model_optimization/python/core/quantization/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_preserve_quantize_registry.py 225 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_utils.py 59 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/default_8bit_cluster_preserve_quantize_scheme.py 10 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/default_8bit_prune_preserve_quantize_scheme.py 9 tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve/prune_preserve_quantize_registry.py 128 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_configs.py 29 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_layout_transform.py 35 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_registry.py 308 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_scheme.py 13 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantizers.py 28 tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_transforms.py 422 tensorflow_model_optimization/python/core/quantization/keras/experimental/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantize_configs.py 37 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantize_layout_transform.py 67 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantize_registry.py 361 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantize_scheme.py 19 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantizers.py 38 tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_transforms.py 461 tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py 331 tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/transforms.py 78 tensorflow_model_optimization/python/core/quantization/keras/layers/__init__.py 1 tensorflow_model_optimization/python/core/quantization/keras/layers/conv_batchnorm_test_utils.py 128 tensorflow_model_optimization/python/core/quantization/keras/layers/dense_batchnorm_test_utils.py 31 tensorflow_model_optimization/python/core/quantization/keras/quant_ops.py 184 tensorflow_model_optimization/python/core/quantization/keras/quantize.py 194 tensorflow_model_optimization/python/core/quantization/keras/quantize_annotate.py 68 tensorflow_model_optimization/python/core/quantization/keras/quantize_aware_activation.py 86 tensorflow_model_optimization/python/core/quantization/keras/quantize_config.py 41 tensorflow_model_optimization/python/core/quantization/keras/quantize_layer.py 52 tensorflow_model_optimization/python/core/quantization/keras/quantize_layout_transform.py 10 tensorflow_model_optimization/python/core/quantization/keras/quantize_registry.py 7 tensorflow_model_optimization/python/core/quantization/keras/quantize_scheme.py 10 tensorflow_model_optimization/python/core/quantization/keras/quantize_wrapper.py 137 tensorflow_model_optimization/python/core/quantization/keras/quantizers.py 150 tensorflow_model_optimization/python/core/quantization/keras/utils.py 38 tensorflow_model_optimization/python/core/sparsity/__init__.py 1 tensorflow_model_optimization/python/core/sparsity/keras/__init__.py 1 tensorflow_model_optimization/python/core/sparsity/keras/estimator_utils.py 48 tensorflow_model_optimization/python/core/sparsity/keras/prunable_layer.py 7 tensorflow_model_optimization/python/core/sparsity/keras/prune.py 83 tensorflow_model_optimization/python/core/sparsity/keras/prune_registry.py 164 tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py 74 tensorflow_model_optimization/python/core/sparsity/keras/pruning_impl.py 156 tensorflow_model_optimization/python/core/sparsity/keras/pruning_policy.py 146 tensorflow_model_optimization/python/core/sparsity/keras/pruning_schedule.py 115 tensorflow_model_optimization/python/core/sparsity/keras/pruning_utils.py 164 tensorflow_model_optimization/python/core/sparsity/keras/pruning_wrapper.py 192 tensorflow_model_optimization/python/core/sparsity/keras/tools/__init__.py 1 tensorflow_model_optimization/python/core/sparsity/keras/tools/check_sparsity_m_by_n.py 50 tensorflow_model_optimization/python/core/sparsity/keras/tools/evaluate_pruning.py 80 tensorflow_model_optimization/python/core/sparsity/keras/tools/sparsity_tooling.py 37 tensorflow_model_optimization/python/core/version.py 11 tensorflow_model_optimization/tensorflow_model_optimization.bzl 33