Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc165 const TfLiteTensor* cell_to_input_weights = in CheckInputTensorDimensions() local
167 if (cell_to_input_weights != nullptr) { in CheckInputTensorDimensions()
168 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckInputTensorDimensions()
169 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
189 ((cell_to_input_weights != nullptr || use_cifg) && in CheckInputTensorDimensions()
192 ((cell_to_input_weights == nullptr) && in CheckInputTensorDimensions()
462 const TfLiteTensor* cell_to_input_weights = in Eval() local
505 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in Eval()
537 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in Eval()
Dlstm_eval.h35 const TfLiteTensor* cell_to_input_weights,
64 const TfLiteTensor* cell_to_input_weights,
Dlstm.cc183 const TfLiteTensor* cell_to_input_weights = in CheckInputTensorDimensions() local
185 if (cell_to_input_weights) { in CheckInputTensorDimensions()
186 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckInputTensorDimensions()
187 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
206 ((cell_to_input_weights != nullptr || use_cifg) && in CheckInputTensorDimensions()
209 ((cell_to_input_weights == nullptr) && in CheckInputTensorDimensions()
530 const TfLiteTensor* cell_to_input_weights = in Eval() local
585 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in Eval()
616 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in Eval()
Dbidirectional_sequence_lstm.cc252 const TfLiteTensor* cell_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
254 if (cell_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
255 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
256 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
257 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
282 ((cell_to_input_weights != nullptr || use_cifg) && in CheckLstmTensorDimensionsAndTypes()
285 ((cell_to_input_weights == nullptr) && in CheckLstmTensorDimensionsAndTypes()
Dlstm_eval.cc879 const TfLiteTensor* cell_to_input_weights, in EvalFloat() argument
945 (use_peephole && !use_cifg) ? cell_to_input_weights->data.f : nullptr; in EvalFloat()
1077 const TfLiteTensor* cell_to_input_weights, in EvalHybrid() argument
1164 GetInt8DataPtr(cell_to_input_weights, is_uint8_hybrid); in EvalHybrid()
1165 cell_to_input_weights_scale = cell_to_input_weights->params.scale; in EvalHybrid()