Searched refs:seq_len_max (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 143 def _block_lstm(seq_len_max, argument
214 seq_len_max=seq_len_max,
301 seq_len_max, x, cs_prev, h_prev, w, wci, wcf, wco, b = op.inputs
309 seq_len_max,
700 seq_len_max=max_seq_len,
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 959 const int64 seq_len_max = seq_len_max_tensor->scalar<int64>()(); in Compute() local
961 for (int64 t = 0; t < seq_len_max; ++t) { in Compute()
990 if (seq_len_max < timelen) { in Compute()
991 Tensor cs_tensor = cs_out->Slice(seq_len_max, timelen); in Compute()
992 Tensor h_tensor = h_out->Slice(seq_len_max, timelen); in Compute()
1217 const int64 seq_len_max = seq_len_max_tensor->scalar<int64>()(); in Compute() local
1219 for (int64 t = seq_len_max - 1; t >= 0; --t) { in Compute()
1271 if (seq_len_max < timelen) { in Compute()
1272 Tensor x_grad_tensor = x_grad->Slice(seq_len_max, timelen); in Compute()
|