tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py
tensorflow_model_optimization/python/core/quantization/keras/quantize_wrapper.py

tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py
tensorflow_model_optimization/python/core/quantization/keras/quantize_annotate.py