from tensorflow.python.ops.rnn_cell_impl import _Linear

    if self._candidate_linear is None:
      with vs.variable_scope("candidate"):
        self._candidate_linear = _Linear(
            [inputs, r_state],
