Searched refs:cell_to_input_weights (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dlstm.cc139 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 Dunidirectional_sequence_lstm.cc139 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