Home
last modified time | relevance | path

Searched refs:TfLiteLSTMParams (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h122 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,
Dlstm_eval.cc874 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepFloat()
1080 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepHybrid()
1673 const int32_t* projection_bias_ptr, const TfLiteLSTMParams* params, in LstmStepInteger8x8_8()
1769 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
1965 const TfLiteTensor* projection_bias, const TfLiteLSTMParams* params, in EvalHybrid()
2233 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalInteger8x8_16()
2471 const TfLiteLSTMParams* params, TfLiteTensor* output_state, in EvalInteger8x8_8()
Dlstm.cc146 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in PopulateQuantizedLstmParams8x8_16()
808 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in PopulateQuantizedLstmParams8x8_8()
857 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
1830 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in Eval()
2355 const auto* params = reinterpret_cast<const TfLiteLSTMParams*>(buffer); in Init()
Dunidirectional_sequence_lstm.cc421 const auto* params = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in CheckInputTensorDimensions()
1321 TfLiteLSTMParams lstm_params; in Eval()
Dlstm_eval_test.cc649 const TfLiteLSTMParams GetLSTMParam() { in GetLSTMParam()
Dbidirectional_sequence_lstm.cc1148 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.h29 GraphFloat32* graph, ObjectReader* reader, const TfLiteLSTMParams* params,
Dlstm_parser.cc463 GraphFloat32* graph, ObjectReader* reader, const TfLiteLSTMParams* params, in ParseLSTMAttributes()
Dmodel_builder.cc1353 const TfLiteLSTMParams* tf_options; in Parse()
1372 const TfLiteLSTMParams* tf_options) { in ParseBasic()
1423 absl::Status CheckBasicParameters(const TfLiteLSTMParams* tf_options) { in CheckBasicParameters()
1439 const TfLiteLSTMParams* tf_options) { in ParseFull()
1447 absl::Status CheckFullParameters(const TfLiteLSTMParams* tf_options) { in CheckFullParameters()
Dmodel_builder_test.cc1963 TfLiteLSTMParams* tf_options = in TEST()
1964 static_cast<TfLiteLSTMParams*>(context->node()->builtin_data); in TEST()
1974 tf_options = static_cast<TfLiteLSTMParams*>(context->node()->builtin_data); in TEST()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc52 TfLiteLSTMParams lstm_params; in TEST()
Dbuiltin_op_data.h276 } TfLiteLSTMParams; typedef
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc181 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepCalibration()
288 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalCalibration()
581 TfLiteLSTMParams lstm_params; in lstm_eval()
585 lstm_params = *(static_cast<TfLiteLSTMParams*>(node->builtin_data)); in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc63 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput()
364 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
546 const auto* params = static_cast<TfLiteLSTMParams*>(node->builtin_data); in lstm_eval()
/external/armnn/delegate/classic/src/
DLstm.hpp37 const auto nodeParams = reinterpret_cast<TfLiteLSTMParams*>(tfLiteNode->builtin_data); in VisitLstmOperator()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc305 reinterpret_cast<TfLiteLSTMParams*>(op_sig.builtin_data); in GetBuiltinOperatorVersion()
Dgpu_compatibility.cc571 const TfLiteLSTMParams* tf_options; in CheckGpuDelegateCompatibility()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1626 auto params = safe_allocator.Allocate<TfLiteLSTMParams>(); in ParseLSTM()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc844 TfLiteLSTMParams lstm_params = {}; in GetVersion()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc4184 auto builtin = reinterpret_cast<TfLiteLSTMParams*>( in Map()
5604 auto builtin = reinterpret_cast<TfLiteLSTMParams*>(node->builtin_data); in AddOpsAndTensors()