| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_eval.h | 37 const TfLiteTensor* cell_to_output_weights,
66 const TfLiteTensor* cell_to_output_weights,
|
| unidirectional_sequence_lstm.cc | 179 const TfLiteTensor* cell_to_output_weights = local
181 if (cell_to_output_weights != nullptr) {
182 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1);
183 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell);
191 (cell_to_output_weights != nullptr)) ||
194 (cell_to_output_weights == nullptr));
466 const TfLiteTensor* cell_to_output_weights = local
505 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights,
537 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights,
|
| lstm.cc | 197 const TfLiteTensor* cell_to_output_weights = local
199 if (cell_to_output_weights) {
200 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1);
201 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell);
208 (cell_to_output_weights != nullptr)) ||
211 (cell_to_output_weights == nullptr));
534 const TfLiteTensor* cell_to_output_weights = local
585 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights,
616 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights,
[all...] |
| bidirectional_sequence_lstm.cc | 270 const TfLiteTensor* cell_to_output_weights = local
272 if (cell_to_output_weights != nullptr) {
273 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1);
274 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell);
275 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->type,
284 (cell_to_output_weights != nullptr)) ||
287 (cell_to_output_weights == nullptr));
[all...] |
| lstm_eval.cc | 66 // cell_to_output_weights - optional
387 // cell_to_output_weights - optional
[all...] |
