Searched refs:lstm_params (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 493 TfLiteLSTMParams lstm_params; in Eval() local
494 lstm_params.activation = params->activation; in Eval()
495 lstm_params.cell_clip = params->cell_clip; in Eval()
496 lstm_params.proj_clip = params->proj_clip; in Eval()
516 projection_bias, &lstm_params, /*forward_sequence=*/true, time_major, in Eval()
548 projection_bias, &lstm_params, /*forward_sequence=*/true, time_major, in Eval()
|
| D | bidirectional_sequence_lstm.cc | 881 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval() local
926 &lstm_params, in Eval()
946 &lstm_params, in Eval()
990 &lstm_params, in Eval()
1013 &lstm_params, in Eval()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 386 if (auto* lstm_params = op->builtin_options_as_LSTMOptions()) { in ParseOpData() local
388 parse_activation(lstm_params->fused_activation_function()); in ParseOpData()
389 params->cell_clip = lstm_params->cell_clip(); in ParseOpData()
390 params->proj_clip = lstm_params->proj_clip(); in ParseOpData()
391 switch (lstm_params->kernel_type()) { in ParseOpData()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data_test.cc | 52 TfLiteLSTMParams lstm_params; in TEST() local
|