Searched refs:TfLiteLSTMParams (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 50 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
79 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
|
| D | lstm.cc | 120 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
506 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in Eval()
837 const auto* params = reinterpret_cast<const TfLiteLSTMParams*>(buffer); in Init()
|
| D | lstm_eval.cc | 116 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
466 const float* projection_bias_ptr, const TfLiteLSTMParams* params, in LstmStepWithAuxInput()
894 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
1092 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalHybrid()
|
| D | unidirectional_sequence_lstm.cc | 103 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
493 TfLiteLSTMParams lstm_params; in Eval()
|
| D | bidirectional_sequence_lstm.cc | 881 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data_test.cc | 52 TfLiteLSTMParams lstm_params; in TEST()
|
| D | builtin_op_data.h | 204 } TfLiteLSTMParams; typedef
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 385 auto params = allocator->AllocatePOD<TfLiteLSTMParams>(); in ParseOpData()
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | nnapi_delegate.cc | 388 auto builtin = reinterpret_cast<TfLiteLSTMParams*>(data); in AddOpsAndParams()
|
| /external/tensorflow/tensorflow/lite/ |
| D | nnapi_delegate.cc | 358 auto builtin = reinterpret_cast<TfLiteLSTMParams*>(data); in AddOpsAndParams()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 915 auto builtin = reinterpret_cast<TfLiteLSTMParams*>( in Map()
|