Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc198 const TfLiteTensor* input_gate_bias = in CheckInputTensorDimensions() local
201 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr); in CheckInputTensorDimensions()
203 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
204 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
469 const TfLiteTensor* input_gate_bias = in Eval() local
514 /*aux_input_to_output_weights=*/nullptr, input_gate_bias, in Eval()
546 /*aux_input_to_output_weights=*/nullptr, input_gate_bias, in Eval()
Dlstm_eval.h47 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
76 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
Dlstm.cc215 const TfLiteTensor* input_gate_bias = in CheckInputTensorDimensions() local
218 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr); in CheckInputTensorDimensions()
220 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
221 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
554 const TfLiteTensor* input_gate_bias = in Eval() local
592 /*aux_input_to_output_weights=*/nullptr, input_gate_bias, in Eval()
623 /*aux_input_to_output_weights=*/nullptr, input_gate_bias, in Eval()
Dbidirectional_sequence_lstm.cc291 const TfLiteTensor* input_gate_bias = in CheckLstmTensorDimensionsAndTypes() local
294 TF_LITE_ENSURE_EQ(context, input_gate_bias, nullptr); in CheckLstmTensorDimensionsAndTypes()
296 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
297 TF_LITE_ENSURE_EQ(context, input_gate_bias->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
298 TF_LITE_ENSURE_EQ(context, input_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
Dlstm_eval.cc891 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() argument
943 (use_cifg) ? nullptr : input_gate_bias->data.f; in EvalFloat()
1089 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalHybrid() argument
1150 input_gate_bias_ptr = input_gate_bias->data.f; in EvalHybrid()