Searched refs:TfLiteLSTMParams (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 122 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
159 const TfLiteTensor* projection_bias, const TfLiteLSTMParams* params,
191 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
217 const TfLiteLSTMParams* params, TfLiteTensor* output_state,
|
| D | lstm_eval.cc | 831 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepFloat()
1024 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepHybrid()
1615 const int32_t* projection_bias_ptr, const TfLiteLSTMParams* params, in LstmStepInteger8x8_8()
1711 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
1901 const TfLiteTensor* projection_bias, const TfLiteLSTMParams* params, in EvalHybrid()
2169 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalInteger8x8_16()
2407 const TfLiteLSTMParams* params, TfLiteTensor* output_state, in EvalInteger8x8_8()
|
| D | lstm.cc | 145 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in PopulateQuantizedLstmParams8x8_16()
807 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in PopulateQuantizedLstmParams8x8_8()
856 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
1826 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in Eval()
2384 const auto* params = reinterpret_cast<const TfLiteLSTMParams*>(buffer); in Init()
|
| D | unidirectional_sequence_lstm.cc | 413 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
1308 TfLiteLSTMParams lstm_params; in Eval()
|
| D | lstm_eval_test.cc | 649 const TfLiteLSTMParams GetLSTMParam() { in GetLSTMParam()
|
| D | bidirectional_sequence_lstm.cc | 1144 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.h | 29 GraphFloat32* graph, ObjectReader* reader, const TfLiteLSTMParams* params,
|
| D | model_builder.cc | 903 const TfLiteLSTMParams* tf_options; in IsSupported()
932 const TfLiteLSTMParams* tf_options; in Parse()
951 const TfLiteLSTMParams* tf_options) { in ParseBasic()
1002 absl::Status CheckBasicParameters(const TfLiteLSTMParams* tf_options) { in CheckBasicParameters()
1018 const TfLiteLSTMParams* tf_options) { in ParseFull()
1026 absl::Status CheckFullParameters(const TfLiteLSTMParams* tf_options) { in CheckFullParameters()
|
| D | lstm_parser.cc | 445 GraphFloat32* graph, ObjectReader* reader, const TfLiteLSTMParams* params, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data_test.cc | 52 TfLiteLSTMParams lstm_params; in TEST()
|
| D | builtin_op_data.h | 275 } TfLiteLSTMParams; typedef
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 180 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
287 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalCalibration()
580 TfLiteLSTMParams lstm_params; in lstm_eval()
584 lstm_params = *(static_cast<TfLiteLSTMParams*>(node->builtin_data)); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
355 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
534 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 562 auto params = safe_allocator.Allocate<TfLiteLSTMParams>(); in ParseOpDataTfLite()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3355 auto builtin = reinterpret_cast<TfLiteLSTMParams*>( in Map()
4256 auto builtin = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in AddOpsAndTensors()
|