Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h120 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias,
156 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias,
189 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias,
215 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias,
Dunidirectional_sequence_lstm.cc566 const TfLiteTensor* output_gate_bias; in CheckInputTensorDimensions() local
569 &output_gate_bias)); in CheckInputTensorDimensions()
570 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
571 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
573 TF_LITE_ENSURE_TYPES_EQ(context, output_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
575 TF_LITE_ENSURE_TYPES_EQ(context, output_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
806 const TfLiteTensor* output_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
813 context, input_zero_point, input_to_output_weights, output_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
1268 const TfLiteTensor* output_gate_bias; in Eval() local
1271 &output_gate_bias)); in Eval()
[all …]
Dlstm.cc523 const TfLiteTensor* output_gate_bias; in PopulateQuantizedLstmParams8x8_8() local
525 &output_gate_bias)); in PopulateQuantizedLstmParams8x8_8()
665 output_gate_bias_ptr = output_gate_bias->data.i32; in PopulateQuantizedLstmParams8x8_8()
1017 const TfLiteTensor* output_gate_bias; in CheckInputTensorDimensions() local
1019 &output_gate_bias)); in CheckInputTensorDimensions()
1020 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
1021 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
1023 TF_LITE_ENSURE_TYPES_EQ(context, output_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
1025 TF_LITE_ENSURE_TYPES_EQ(context, output_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
1244 const TfLiteTensor* output_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
Dlstm_eval.cc1709 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias, in EvalFloat() argument
1793 GetTensorData<float>(output_gate_bias), in EvalFloat()
1854 GetTensorData<float>(output_gate_bias), in EvalFloat()
1898 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias, in EvalHybrid()
2024 GetTensorData<float>(output_gate_bias), in EvalHybrid()
2122 GetTensorData<float>(output_gate_bias), in EvalHybrid()
2167 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias, in EvalInteger8x8_16() argument
2260 GetTensorData<int32_t>(output_gate_bias), in EvalInteger8x8_16()
2359 GetTensorData<int32_t>(output_gate_bias), in EvalInteger8x8_16()
2405 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias, in EvalInteger8x8_8() argument
[all …]
Dlstm_test.cc1626 const std::vector<float> output_gate_bias = {0.05, -0.01, 0.2, 0.1}; in TEST() local
1711 lstm.SetOutputGateBias(output_gate_bias); in TEST()
1788 const std::vector<float> output_gate_bias = {0.05, -0.01, 0.2, 0.1}; in TEST() local
1880 lstm.SetOutputGateBias(output_gate_bias); in TEST()
1961 const std::vector<float> output_gate_bias = {0.05, -0.01, 0.2, 0.1}; in TEST() local
2050 lstm.SetOutputGateBias(output_gate_bias); in TEST()
Dbidirectional_sequence_lstm.cc345 const TfLiteTensor* output_gate_bias; in CheckLstmTensorDimensionsAndTypes() local
348 GetInputSafe(context, node, output_gate_bias_tensor, &output_gate_bias)); in CheckLstmTensorDimensionsAndTypes()
349 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
350 TF_LITE_ENSURE_EQ(context, output_gate_bias->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
351 TF_LITE_ENSURE_TYPES_EQ(context, output_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc3062 const std::vector<float> output_gate_bias = {0.05, -0.01, 0.2, 0.1}; in TEST() local
3147 lstm.SetOutputGateBias(output_gate_bias); in TEST()
3220 const std::vector<float> output_gate_bias = {0.05, -0.01, 0.2, 0.1}; in TEST() local
3312 lstm.SetOutputGateBias(output_gate_bias); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc285 const TfLiteTensor* cell_gate_bias, const TfLiteTensor* output_gate_bias, in EvalCalibration() argument
371 GetTensorData<float>(output_gate_bias), in EvalCalibration()
433 GetTensorData<float>(output_gate_bias), in EvalCalibration()
545 const TfLiteTensor* output_gate_bias; in lstm_eval() local
549 &output_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.cc353 const TfLiteTensor* cell_bias, const TfLiteTensor* output_gate_bias, in EvalFloat() argument
440 GetTensorData<float>(output_gate_bias), in EvalFloat()
502 GetTensorData<float>(output_gate_bias), in EvalFloat()
583 const TfLiteTensor* output_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.cc54 std::initializer_list<float> output_gate_bias, in LSTMOpModel() argument
109 AddConstInput({TensorType_FLOAT32, {n_cell}}, output_gate_bias); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc183 lstm.output_layer_norm_coefficients(), lstm.output_gate_bias(), builder); in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3806 TFL_OperandHasRank<15, 1>, // output_gate_bias
3857 TFL_TensorOf<[F32, QI32]>:$output_gate_bias,
3938 TFL_OperandHasRank<15, 1>, // output_gate_bias
3981 TFL_FpTensor:$output_gate_bias,