Home
last modified time | relevance | path

Searched refs:input_to_output_weights (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc270 const TfLiteTensor* input_to_output_weights = in Prepare() local
272 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare()
273 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in Prepare()
274 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in Prepare()
309 const bool is_hybrid_op = ((input_to_output_weights->type == kTfLiteUInt8 || in Prepare()
310 input_to_output_weights->type == kTfLiteInt8) && in Prepare()
348 input_quantized->type = input_to_output_weights->type; in Prepare()
359 activation_state_quantized->type = input_to_output_weights->type; in Prepare()
373 cell_state_quantized->type = input_to_output_weights->type; in Prepare()
450 const TfLiteTensor* input_to_output_weights = in Eval() local
[all …]
Dlstm.cc344 const TfLiteTensor* input_to_output_weights = in Prepare() local
346 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare()
347 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in Prepare()
348 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in Prepare()
385 const bool is_hybrid_op = ((input_to_output_weights->type == kTfLiteUInt8 || in Prepare()
386 input_to_output_weights->type == kTfLiteInt8) && in Prepare()
422 input_quantized->type = input_to_output_weights->type; in Prepare()
432 activation_state_quantized->type = input_to_output_weights->type; in Prepare()
445 cell_state_quantized->type = input_to_output_weights->type; in Prepare()
518 const TfLiteTensor* input_to_output_weights = in Eval() local
[all …]
Dlstm_eval.h30 const TfLiteTensor* input_to_output_weights,
59 const TfLiteTensor* input_to_output_weights,
Dlstm_eval.cc874 const TfLiteTensor* input_to_output_weights, in EvalFloat() argument
912 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat()
998 input_to_output_weights->data.f, aux_input_ptr, in EvalFloat()
1044 input_to_output_weights->data.f, aux_input_ptr, in EvalFloat()
1072 const TfLiteTensor* input_to_output_weights, in EvalHybrid() argument
1101 const bool is_uint8_hybrid = input_to_output_weights->type == kTfLiteUInt8; in EvalHybrid()
1115 const int n_cell = input_to_output_weights->dims->data[0]; in EvalHybrid()
1203 GetInt8DataPtr(input_to_output_weights, is_uint8_hybrid); in EvalHybrid()
1205 input_to_output_weights->params.scale; in EvalHybrid()
Dbidirectional_sequence_lstm.cc204 const TfLiteTensor* input_to_output_weights = in CheckLstmTensorDimensionsAndTypes() local
206 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
207 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
208 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
209 TF_LITE_ENSURE_EQ(context, input_to_output_weights->type, in CheckLstmTensorDimensionsAndTypes()