Home
last modified time | relevance | path

Searched refs:forget_gate_bias (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h119 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
155 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
188 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
214 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
Dunidirectional_sequence_lstm.cc542 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local
545 &forget_gate_bias)); in CheckInputTensorDimensions()
546 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
547 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
549 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
551 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
774 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
781 context, input_zero_point, input_to_forget_weights, forget_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
1260 const TfLiteTensor* forget_gate_bias; in Eval() local
1263 &forget_gate_bias)); in Eval()
[all …]
Dlstm.cc517 const TfLiteTensor* forget_gate_bias; in PopulateQuantizedLstmParams8x8_8() local
519 &forget_gate_bias)); in PopulateQuantizedLstmParams8x8_8()
663 forget_gate_bias_ptr = forget_gate_bias->data.i32; in PopulateQuantizedLstmParams8x8_8()
995 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local
997 &forget_gate_bias)); in CheckInputTensorDimensions()
998 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
999 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
1001 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
1003 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
1215 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
Dlstm_eval.cc1708 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() argument
1791 GetTensorData<float>(forget_gate_bias), in EvalFloat()
1852 GetTensorData<float>(forget_gate_bias), in EvalFloat()
1897 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalHybrid()
2022 GetTensorData<float>(forget_gate_bias), in EvalHybrid()
2120 GetTensorData<float>(forget_gate_bias), in EvalHybrid()
2166 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_16() argument
2258 GetTensorData<int32_t>(forget_gate_bias), in EvalInteger8x8_16()
2357 GetTensorData<int32_t>(forget_gate_bias), in EvalInteger8x8_16()
2404 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_8() argument
[all …]
Dlstm_test.cc1622 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
1710 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
1784 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
1879 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
1957 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
2049 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
Dbidirectional_sequence_lstm.cc330 const TfLiteTensor* forget_gate_bias; in CheckLstmTensorDimensionsAndTypes() local
333 GetInputSafe(context, node, forget_gate_bias_tensor, &forget_gate_bias)); in CheckLstmTensorDimensionsAndTypes()
334 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
335 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
336 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc3058 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
3146 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
3216 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
3311 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc284 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalCalibration() argument
369 GetTensorData<float>(forget_gate_bias), in EvalCalibration()
431 GetTensorData<float>(forget_gate_bias), in EvalCalibration()
535 const TfLiteTensor* forget_gate_bias; in lstm_eval() local
539 &forget_gate_bias)); in lstm_eval()
620 forget_gate_bias, cell_gate_bias, output_gate_bias, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc352 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() argument
438 GetTensorData<float>(forget_gate_bias), in EvalFloat()
500 GetTensorData<float>(forget_gate_bias), in EvalFloat()
579 const TfLiteTensor* forget_gate_bias = in lstm_eval() local
624 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc52 std::initializer_list<float> forget_gate_bias, in LSTMOpModel() argument
107 AddConstInput({TensorType_FLOAT32, {n_cell}}, forget_gate_bias); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc163 lstm.forget_layer_norm_coefficients(), lstm.forget_gate_bias(), builder); in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3804 TFL_OperandHasRank<13, 1>, // forget_gate_bias
3855 TFL_TensorOf<[F32, QI32]>:$forget_gate_bias,
3936 TFL_OperandHasRank<13, 1>, // forget_gate_bias
3979 TFL_FpTensor:$forget_gate_bias,