Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc125 const TfLiteTensor* input_to_cell_weights = in CheckInputTensorDimensions() local
127 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
128 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
129 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
448 const TfLiteTensor* input_to_cell_weights = in Eval() local
502 input_to_cell_weights, input_to_output_weights, in Eval()
534 input_to_cell_weights, input_to_output_weights, in Eval()
Dlstm_eval.h29 const TfLiteTensor* input_to_cell_weights,
58 const TfLiteTensor* input_to_cell_weights,
Dlstm.cc143 const TfLiteTensor* input_to_cell_weights = in CheckInputTensorDimensions() local
145 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
146 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
147 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
516 const TfLiteTensor* input_to_cell_weights = in Eval() local
582 input_to_cell_weights, input_to_output_weights, in Eval()
613 input_to_cell_weights, input_to_output_weights, in Eval()
Dbidirectional_sequence_lstm.cc196 const TfLiteTensor* input_to_cell_weights = in CheckLstmTensorDimensionsAndTypes() local
198 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
199 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
200 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
201 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->type, in CheckLstmTensorDimensionsAndTypes()
Dlstm_eval.cc873 const TfLiteTensor* input_to_cell_weights, in EvalFloat() argument
997 input_to_forget_weights->data.f, input_to_cell_weights->data.f, in EvalFloat()
1043 input_to_forget_weights->data.f, input_to_cell_weights->data.f, in EvalFloat()
1071 const TfLiteTensor* input_to_cell_weights, in EvalHybrid() argument
1200 GetInt8DataPtr(input_to_cell_weights, is_uint8_hybrid); in EvalHybrid()
1201 const float input_to_cell_weights_scale = input_to_cell_weights->params.scale; in EvalHybrid()