Searched refs:cell_to_input_weights (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | lstm.cc | 139 TfLiteTensor* cell_to_input_weights = local 141 if (cell_to_input_weights) { 142 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); 143 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); 163 ((cell_to_input_weights != nullptr || use_cifg) && 166 ((cell_to_input_weights == nullptr) && 328 TfLiteTensor* cell_to_input_weights = local 428 cell_to_input_weights->data.f, n_cell, cell_state->data.f, n_batch,
|
H A D | unidirectional_sequence_lstm.cc | 139 TfLiteTensor* cell_to_input_weights = local 141 if (cell_to_input_weights) { 142 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); 143 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); 163 ((cell_to_input_weights != nullptr || use_cifg) && 166 ((cell_to_input_weights == nullptr) && 330 TfLiteTensor* cell_to_input_weights = local 436 cell_to_input_weights->data.f, n_cell, cell_state->data.f, n_batch,
|
Completed in 153 milliseconds