Searched refs:is_bidi (Results 1 – 2 of 2) sorted by relevance
462 is_bidi = self._direction == CUDNN_RNN_BIDIRECTION467 if is_bidi:473 return tf_wts if not is_bidi else tf_wts * 2
232 def _CreateCudnnCompatibleCanonicalRNN(rnn, inputs, is_bidi=False, scope=None): argument250 if not is_bidi:994 rnn, cell_inputs, is_bidi=True)