Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc131 const TfLiteTensor* recurrent_to_input_weights = in CheckInputTensorDimensions() local
133 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
134 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
135 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
137 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
160 (recurrent_to_input_weights != nullptr)) || in CheckInputTensorDimensions()
162 (recurrent_to_input_weights == nullptr)); in CheckInputTensorDimensions()
453 const TfLiteTensor* recurrent_to_input_weights = in Eval() local
503 recurrent_to_input_weights, recurrent_to_forget_weights, in Eval()
535 recurrent_to_input_weights, recurrent_to_forget_weights, in Eval()
Dlstm.cc149 const TfLiteTensor* recurrent_to_input_weights = in CheckInputTensorDimensions() local
151 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
152 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
153 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
155 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
178 (recurrent_to_input_weights != nullptr)) || in CheckInputTensorDimensions()
180 (recurrent_to_input_weights == nullptr)); in CheckInputTensorDimensions()
521 const TfLiteTensor* recurrent_to_input_weights = in Eval() local
583 recurrent_to_input_weights, recurrent_to_forget_weights, in Eval()
614 recurrent_to_input_weights, recurrent_to_forget_weights, in Eval()
Dlstm_eval.h31 const TfLiteTensor* recurrent_to_input_weights,
60 const TfLiteTensor* recurrent_to_input_weights,
Dbidirectional_sequence_lstm.cc212 const TfLiteTensor* recurrent_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
214 if (recurrent_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
215 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
216 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckLstmTensorDimensionsAndTypes()
218 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckLstmTensorDimensionsAndTypes()
220 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
247 (recurrent_to_input_weights != nullptr)) || in CheckLstmTensorDimensionsAndTypes()
249 (recurrent_to_input_weights == nullptr)); in CheckLstmTensorDimensionsAndTypes()
Dlstm_eval.cc875 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat() argument
941 (use_cifg) ? nullptr : recurrent_to_input_weights->data.f; in EvalFloat()
1073 const TfLiteTensor* recurrent_to_input_weights, in EvalHybrid() argument
1149 GetInt8DataPtr(recurrent_to_input_weights, is_uint8_hybrid); in EvalHybrid()
1152 recurrent_to_input_weights_scale = recurrent_to_input_weights->params.scale; in EvalHybrid()