Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc229 const TfLiteTensor* projection_bias = in CheckInputTensorDimensions() local
231 if (projection_bias != nullptr) { in CheckInputTensorDimensions()
232 TF_LITE_ENSURE_EQ(context, projection_bias->dims->size, 1); in CheckInputTensorDimensions()
233 TF_LITE_ENSURE_EQ(context, projection_bias->dims->data[0], n_output); in CheckInputTensorDimensions()
242 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckInputTensorDimensions()
479 const TfLiteTensor* projection_bias = in Eval() local
516 projection_bias, &lstm_params, /*forward_sequence=*/true, time_major, in Eval()
548 projection_bias, &lstm_params, /*forward_sequence=*/true, time_major, in Eval()
Dlstm_eval.h49 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
78 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
Dlstm.cc246 const TfLiteTensor* projection_bias = in CheckInputTensorDimensions() local
248 if (projection_bias != nullptr) { in CheckInputTensorDimensions()
249 TF_LITE_ENSURE_EQ(context, projection_bias->dims->size, 1); in CheckInputTensorDimensions()
250 TF_LITE_ENSURE_EQ(context, projection_bias->dims->data[0], n_output); in CheckInputTensorDimensions()
259 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckInputTensorDimensions()
564 const TfLiteTensor* projection_bias = in Eval() local
594 projection_bias, params, /*forward_sequence=*/true, in Eval()
625 projection_bias, params, /*forward_sequence=*/true, in Eval()
Dbidirectional_sequence_lstm.cc329 const TfLiteTensor* projection_bias = in CheckLstmTensorDimensionsAndTypes() local
331 if (projection_bias != nullptr) { in CheckLstmTensorDimensionsAndTypes()
332 TF_LITE_ENSURE_EQ(context, projection_bias->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
333 TF_LITE_ENSURE_EQ(context, projection_bias->dims->data[0], n_output); in CheckLstmTensorDimensionsAndTypes()
334 TF_LITE_ENSURE_EQ(context, projection_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
343 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckLstmTensorDimensionsAndTypes()
Dlstm_eval.cc893 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat() argument
962 (projection_bias == nullptr) ? nullptr : projection_bias->data.f; in EvalFloat()
1091 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalHybrid() argument
1192 (projection_bias == nullptr) ? nullptr : projection_bias->data.f; in EvalHybrid()