diff options
Diffstat (limited to 'tensorflow/contrib/quantize/python/common.py')
-rw-r--r-- | tensorflow/contrib/quantize/python/common.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/quantize/python/common.py b/tensorflow/contrib/quantize/python/common.py index b27117dd48..e6c04bcf55 100644 --- a/tensorflow/contrib/quantize/python/common.py +++ b/tensorflow/contrib/quantize/python/common.py @@ -34,10 +34,10 @@ SKIPPED_PREFIXES = ( 'ScalarSummary') # Valid activation ops for quantization end points. -_ACTIVATION_OP_SUFFIXES = ['/Relu6', '/Relu', '/Identity'] +_ACTIVATION_OP_SUFFIXES = ['Relu6', 'Relu', 'Identity'] # Regular expression for recognizing nodes that are part of batch norm group. -_BATCHNORM_RE = re.compile(r'^(.*)/BatchNorm/batchnorm') +_BATCHNORM_RE = re.compile(r'^(.*)BatchNorm/batchnorm') def BatchNormGroups(graph): |