| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_eval.h | 47 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
76 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
|
| unidirectional_sequence_lstm.cc | 198 const TfLiteTensor* input_gate_bias = local
201 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr);
203 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1);
204 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell);
469 const TfLiteTensor* input_gate_bias = local
514 /*aux_input_to_output_weights=*/nullptr, input_gate_bias,
546 /*aux_input_to_output_weights=*/nullptr, input_gate_bias,
|
| lstm.cc | 215 const TfLiteTensor* input_gate_bias = local
218 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr);
220 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1);
221 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell);
554 const TfLiteTensor* input_gate_bias = local
592 /*aux_input_to_output_weights=*/nullptr, input_gate_bias,
623 /*aux_input_to_output_weights=*/nullptr, input_gate_bias,
[all...] |
| bidirectional_sequence_lstm.cc | 291 const TfLiteTensor* input_gate_bias = local
294 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr);
296 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1);
297 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell);
298 TF_LITE_ENSURE_EQ(context, input_gate_bias->type, kTfLiteFloat32);
[all...] |
| lstm_eval.cc | [all...] |
