ejcgt

**Describe the bug** tensorflow_model_optimization.quantization.keras.quantize_model function throws the following error with mobilenet v3. ``` Traceback (most recent call last): File "issue_report.py", line 14, in q_aware_model = tfmo.quantization.keras.quantize_model(model_mv3) File "/home/.local/lib/python3.8/site-packages/tensorflow_model_optimization/python/core/quantization/keras/quantize.py", line 137,...