| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_eval.h | 48 const TfLiteTensor* cell_bias, const TfLiteTensor* output_gate_bias,
77 const TfLiteTensor* cell_bias, const TfLiteTensor* output_gate_bias,
|
| unidirectional_sequence_lstm.cc | 212 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); local
213 TF_LITE_ENSURE_EQ(context, cell_bias->dims->size, 1);
214 TF_LITE_ENSURE_EQ(context, cell_bias->dims->data[0], n_cell);
473 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); local
515 forget_gate_bias, cell_bias, output_gate_bias, projection_weights,
547 forget_gate_bias, cell_bias, output_gate_bias, projection_weights,
|
| lstm.cc | 229 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); local
230 TF_LITE_ENSURE_EQ(context, cell_bias->dims->size, 1);
231 TF_LITE_ENSURE_EQ(context, cell_bias->dims->data[0], n_cell);
558 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); local
593 forget_gate_bias, cell_bias, output_gate_bias, projection_weights,
624 forget_gate_bias, cell_bias, output_gate_bias, projection_weights,
[all...] |
| lstm_eval.cc | [all...] |
| bidirectional_sequence_lstm.cc | 307 const TfLiteTensor* cell_bias = local
309 TF_LITE_ENSURE_EQ(context, cell_bias->dims->size, 1);
310 TF_LITE_ENSURE_EQ(context, cell_bias->dims->data[0], n_cell);
311 TF_LITE_ENSURE_EQ(context, cell_bias->type, kTfLiteFloat32);
[all...] |
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn_cell.py | 354 self.cell_bias = add_variable_wrapped(
355 "cell_bias", bias_shape, bias_initializer, 14, maybe_partitioner)
474 transpose_b=True), self.cell_bias)
|
