Searched refs:recurrent_to_cell_weights (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | lstm.cc | 123 TfLiteTensor* recurrent_to_cell_weights = local
125 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->size, 2);
126 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[0], n_cell);
127 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[1],
323 TfLiteTensor* recurrent_to_cell_weights = local
418 recurrent_to_cell_weights->data.f, n_cell, n_output, output_state->data.f,
|
| H A D | unidirectional_sequence_lstm.cc | 123 TfLiteTensor* recurrent_to_cell_weights = local
125 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->size, 2);
126 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[0], n_cell);
127 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[1],
325 TfLiteTensor* recurrent_to_cell_weights = local
425 recurrent_to_cell_weights->data.f, n_cell, n_output,
|
Completed in 69 milliseconds