102b6a5cdb85acd490e23831b643fc679cbb8db5,graph_compression/compression_lib/simhash_compression_op.py,KMeansPruningCompressionOp,get_customized_apply_compression_op,#KMeansPruningCompressionOp#,851

Before Change

                                               layer_obj.vars.mask,
                                               "masked_weight")

      if self._spec.update_option in [0, 2]:
        self.update_op = self._create_update_op()
      else:
        self.update_op = self.setup_update_explicit()

After Change

                tf.less(self._spec.end_compression_step, 0)))
        return is_step_within_compression_range

    if getattr(self._spec, "do_transpose", False):
      self.pruning_and_compression_op = (
          self.alpha * self.pruned_a_matrix_tfvar +
          (1 - self.alpha) * tf.math.multiply(

Instances