def _compressor_and_alpha_update_op(self):
    """Applies compressor and also updates alpha."""

    self._compressor_op(self.matrix_compressor, self.a_matrix_tfvar)
    self._update_alpha_op()
    self.last_alpha_update_step.assign(tf.cast(self._global_step, tf.int32))