Searched refs:timelen (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 188 DimensionHandle timelen = c->Dim(x, 0); in __anon652ace880502() local
195 ShapeHandle output = c->MakeShape({timelen, batch_size, cell_size}); in __anon652ace880502()
227 DimensionHandle timelen = c->Dim(x, 0); in __anon652ace880602() local
234 ShapeHandle output = c->MakeShape({timelen, batch_size, cell_size}); in __anon652ace880602()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 849 const int64 timelen = x->dim_size(0); in Compute() local
936 TensorShape batch_cell_shape({timelen, batch_size, cell_size}); in Compute()
1003 if (seq_len_max < timelen) { in Compute()
1004 Tensor cs_tensor = cs_out->Slice(seq_len_max, timelen); in Compute()
1005 Tensor h_tensor = h_out->Slice(seq_len_max, timelen); in Compute()
1083 const int64 timelen = x->dim_size(0); in Compute() local
1144 TensorShape batch_input_shape({timelen, batch_size, input_size}); in Compute()
1291 if (seq_len_max < timelen) { in Compute()
1292 Tensor x_grad_tensor = x_grad->Slice(seq_len_max, timelen); in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BlockLSTM.pbtxt | 14 The sequence input to the LSTM, shape (timelen, batch_size, num_inputs).
|
| D | api_def_BlockLSTMGradV2.pbtxt | 14 The sequence input to the LSTM, shape (timelen, batch_size, num_inputs).
|
| D | api_def_BlockLSTMGrad.pbtxt | 14 The sequence input to the LSTM, shape (timelen, batch_size, num_inputs).
|
| D | api_def_BlockLSTMV2.pbtxt | 14 The sequence input to the LSTM, shape (timelen, batch_size, num_inputs).
|