Change comment to indicate the difference

tensorflow · Sep 27, 2020 · 2d0aa42 · 2d0aa42
1 parent b3d640c
commit 2d0aa42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...el_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py b/...el_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py
@@ -416,7 +416,7 @@ def _add_replacement_nodes(first_layer_removed_index, replacement_nodes):
   def _weight_name(name):
     """Extracts the weight name by removing layer from TF variable name.
 
-    For example, returns 'kernel:0' for 'dense_2/kernel:0'.
+    For example, returns 'kernel:0' for 'quantize_annotate_2/conv1d_25/kernel:0'.
 
     Args:
       name: TensorFlow variable name.