Searched refs:attn_length (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | models.py | 300 attn_length, attn_size, attn_vec_size): argument
349 if attn_length is not None:
353 attn_length=attn_length,
361 attn_length=attn_length,
386 if attn_length is not None:
390 attn_length=attn_length,
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/ |
| D | seq2seq.py | 616 attn_length = attention_states.get_shape()[1].value
617 if attn_length is None:
618 attn_length = array_ops.shape(attention_states)[1]
623 [-1, attn_length, 1, attn_size])
660 array_ops.reshape(a, [-1, attn_length, 1, 1]) * hidden, [1, 2])
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 802 attn_length = 16
813 lstm_cell, attn_length, state_is_tuple=state_is_tuple)
817 [batch_size, attn_length * num_units], dtype=np.float32)
823 num_units * 2 + attn_length * num_units + num_units
839 [batch_size, attn_length * num_units])
844 num_units * 2 + num_units + attn_length * num_units
854 attn_length = 16
864 lstm_cell, attn_length, state_is_tuple=state_is_tuple)
871 [batch_size, attn_length * num_units], dtype=np.float32),
879 num_units * 2 + num_units + attn_length * num_units
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 1123 attn_length, argument
1160 if attn_length <= 0:
1162 "attn_length should be greater than zero, got %s" % str(attn_length))
1175 self._attn_length = attn_length
|