Searched refs:lstm_params (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 580 TfLiteLSTMParams lstm_params; in lstm_eval() local
584 lstm_params = *(static_cast<TfLiteLSTMParams*>(node->builtin_data)); in lstm_eval()
593 lstm_params.activation = params->activation; in lstm_eval()
594 lstm_params.cell_clip = params->cell_clip; in lstm_eval()
595 lstm_params.proj_clip = params->proj_clip; in lstm_eval()
596 lstm_params.asymmetric_quantize_inputs = in lstm_eval()
621 projection_weights, projection_bias, &lstm_params, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data_test.cc | 52 TfLiteLSTMParams lstm_params; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 1308 TfLiteLSTMParams lstm_params; in Eval() local
1309 lstm_params.activation = params->activation; in Eval()
1310 lstm_params.cell_clip = params->cell_clip; in Eval()
1311 lstm_params.proj_clip = params->proj_clip; in Eval()
1312 lstm_params.asymmetric_quantize_inputs = params->asymmetric_quantize_inputs; in Eval()
1334 projection_weights, projection_bias, &lstm_params, in Eval()
1379 projection_bias, &lstm_params, in Eval()
1426 projection_bias, &lstm_params, /*forward_sequence=*/true, in Eval()
|
| D | bidirectional_sequence_lstm.cc | 1144 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval() local
1190 &lstm_params, in Eval()
1210 &lstm_params, in Eval()
1287 &lstm_params, in Eval()
1328 &lstm_params, in Eval()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 564 if (const auto* lstm_params = op->builtin_options_as_LSTMOptions()) { in ParseOpDataTfLite() local
566 ConvertActivation(lstm_params->fused_activation_function()); in ParseOpDataTfLite()
567 params->cell_clip = lstm_params->cell_clip(); in ParseOpDataTfLite()
568 params->proj_clip = lstm_params->proj_clip(); in ParseOpDataTfLite()
569 switch (lstm_params->kernel_type()) { in ParseOpDataTfLite()
579 lstm_params->kernel_type()); in ParseOpDataTfLite()
583 lstm_params->asymmetric_quantize_inputs(); in ParseOpDataTfLite()
|