Searched defs:forget_gate_bias (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_lstm.cc | 542 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local 774 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local 1260 const TfLiteTensor* forget_gate_bias; in Eval() local
|
D | lstm.cc | 517 const TfLiteTensor* forget_gate_bias; in PopulateQuantizedLstmParams8x8_8() local 995 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local 1215 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local 1880 const TfLiteTensor* forget_gate_bias; in Eval() local
|
D | lstm_test.cc | 1622 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local 1784 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local 1957 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
|
D | lstm_eval.cc | 1708 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() 1868 TfLiteStatus EvalHybrid( in EvalHybrid() 2166 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_16() 2404 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_8()
|
D | unidirectional_sequence_lstm_test.cc | 3058 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local 3216 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
|
D | bidirectional_sequence_lstm.cc | 330 const TfLiteTensor* forget_gate_bias; in CheckLstmTensorDimensionsAndTypes() local
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 352 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() 579 const TfLiteTensor* forget_gate_bias = in lstm_eval() local
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
D | lstm.cc | 284 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalCalibration() 535 const TfLiteTensor* forget_gate_bias; in lstm_eval() local
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
D | lstm_full_test.cc | 52 std::initializer_list<float> forget_gate_bias, in LSTMOpModel()
|