Skip to content

Commit d43b950

Browse files
nutsiepullytensorflower-gardener
authored andcommitted
Rename QuantizeWrapper to be more succinct.
PiperOrigin-RevId: 285255767
1 parent f148e43 commit d43b950

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,9 @@ def __init__(self, layer, quantize_provider, **kwargs):
5454
raise ValueError('quantize_provider cannot be None. It is needed to '
5555
'quantize a layer.')
5656

57+
if 'name' not in kwargs:
58+
kwargs['name'] = self._make_layer_name(layer)
59+
5760
super(QuantizeWrapper, self).__init__(layer, **kwargs)
5861
self.quantize_provider = quantize_provider
5962

@@ -63,6 +66,10 @@ def __init__(self, layer, quantize_provider, **kwargs):
6366
self._batch_input_shape = self.layer._batch_input_shape # pylint: disable=protected-access
6467
self._track_trackable(layer, name='layer')
6568

69+
@staticmethod
70+
def _make_layer_name(layer):
71+
return '{}_{}'.format('quant', layer.name)
72+
6673
@staticmethod
6774
def _weight_name(name):
6875
"""Extracts the weight name from the full TensorFlow variable name.

0 commit comments

Comments
 (0)