No public description

PiperOrigin-RevId: 569259447 Change-Id: Ic83bcaf2991e859dbc49f46f92506b5a56d7f49e
google · Sep 28, 2023 · 3d60be2 · 3d60be2
1 parent 51f3265
commit 3d60be2
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 17 deletions.
diff --git a/qkeras/qconvolutional.py b/qkeras/qconvolutional.py
@@ -33,7 +33,7 @@
 from tensorflow.keras.layers import InputSpec
 from tensorflow.python.eager import context
 from tensorflow.python.ops import array_ops
-# Google internal code, commented out by copybara
+# from tensorflow.python.ops import array_ops
 from .qlayers import get_auto_range_constraint_initializer
 from .qlayers import QActivation
 from .quantizers import get_quantized_initializer

diff --git a/qkeras/qrecurrent.py b/qkeras/qrecurrent.py
@@ -32,7 +32,7 @@
 from tensorflow.keras.layers import Bidirectional
 from tensorflow.python.util import nest
 from tensorflow.python.ops import array_ops
-# Google internal code, commented out by copybara
+# from tensorflow.python.ops import array_ops
 from tensorflow.python.framework import ops
 from tensorflow_model_optimization.python.core.sparsity.keras.prunable_layer import PrunableLayer
 

diff --git a/qkeras/qtools/quantized_operators/quantizer_factory.py b/qkeras/qtools/quantized_operators/quantizer_factory.py
@@ -20,8 +20,8 @@
 
 import copy
 from qkeras import quantizers
-# Google internal code, commented out by copybara
-# Google internal code, commented out by copybara
+# from qkeras.google_internals import experimental_quantizers
+# from qkeras.google_internals import experimental_quantizer_impl
 from qkeras.qtools.quantized_operators import quantizer_impl
 from qkeras.qtools.settings import cfg
 
@@ -53,10 +53,10 @@ def __init__(self):
             quantizer_impl.QuantizedTanh,
         quantizers.quantized_ulaw:
             quantizer_impl.QuantizedUlaw,
-        # Google internal code, commented out by copybara
-            # Google internal code, commented out by copybara
-        # Google internal code, commented out by copybara
-            # Google internal code, commented out by copybara
+        # experimental_quantizers.quantized_bits_learnable_scale:
+            # experimental_quantizer_impl.QuantizedBitsLearnableScale,
+        # experimental_quantizers.parametric_quantizer_d_xmax:
+            # experimental_quantizer_impl.ParametricQuantizer,
 
         # add following quantizer types for the use in GraphUpdateEdge
         quantizer_impl.QuantizedBits:
@@ -83,10 +83,10 @@ def __init__(self):
             quantizer_impl.StochasticTernary,
         quantizer_impl.QuantizedUlaw:
             quantizer_impl.StochasticTernary,
-        # Google internal code, commented out by copybara
-            # Google internal code, commented out by copybara
-        # Google internal code, commented out by copybara
-            # Google internal code, commented out by copybara
+        # experimental_quantizer_impl.QuantizedBitsLearnableScale:
+            # experimental_quantizer_impl.QuantizedBitsLearnableScale,
+        #experimental_quantizer_impl.ParametricQuantizer:
+            # experimental_quantizer_impl.ParametricQuantizer,
     }
 
     self._default_interm_quantizer = cfg.default_interm_quantizer

diff --git a/qkeras/quantizers.py b/qkeras/quantizers.py
@@ -28,8 +28,8 @@
 from tensorflow.keras.utils import deserialize_keras_object
 from tensorflow.python.framework import smart_cond as tf_utils
 from .safe_eval import safe_eval
-# Google internal code, commented out by copybara
-# Google internal code, commented out by copybara
+# from .google_internals.experimental_quantizers import quantized_bits_learnable_scale
+# from .google_internals.experimental_quantizers import parametric_quantizer_d_xmax
 
 #
 # Library of auxiliary functions

diff --git a/qkeras/utils.py b/qkeras/utils.py
@@ -70,8 +70,8 @@
 from .quantizers import stochastic_binary
 from .quantizers import stochastic_ternary
 from .quantizers import ternary
-# Google internal code, commented out by copybara
-# Google internal code, commented out by copybara
+# from .google_internals.experimental_quantizers import quantized_bits_learnable_scale
+# from .google_internals.experimental_quantizers import parametric_quantizer_d_xmax
 from .safe_eval import safe_eval
 from tensorflow.python.ops import math_ops
 from .qmac import QScaleShift
@@ -1051,7 +1051,7 @@ def _add_supported_quantized_objects(custom_objects):
   custom_objects["quantized_sigmoid"] = quantized_sigmoid
   custom_objects["quantized_po2"] = quantized_po2
   custom_objects["quantized_relu_po2"] = quantized_relu_po2
-  # Google internal code, commented out by copybara
+  # custom_objects["quantized_bits_learnable_scale"] = quantized_bits_learnable_scale
 
   custom_objects["QConv2DBatchnorm"] = QConv2DBatchnorm
   custom_objects["QDepthwiseConv2DBatchnorm"] = QDepthwiseConv2DBatchnorm