Searched defs:cell_to_forget_weights (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 128 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
496 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1253 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in Eval() local
|
| D | lstm.cc | 213 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_16() local
501 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_8() local
950 const TfLiteTensor* cell_to_forget_weights = in CheckInputTensorDimensions() local
1864 const TfLiteTensor* cell_to_forget_weights = in Eval() local
|
| D | lstm_eval.cc | 1697 const TfLiteTensor* cell_to_forget_weights, in EvalFloat()
1886 const TfLiteTensor* cell_to_forget_weights, in EvalHybrid()
2160 const TfLiteTensor* cell_to_forget_weights, in EvalInteger8x8_16()
2398 const TfLiteTensor* cell_to_forget_weights, in EvalInteger8x8_8()
|
| D | bidirectional_sequence_lstm.cc | 290 const TfLiteTensor* cell_to_forget_weights = in CheckLstmTensorDimensionsAndTypes() local
|
| D | unidirectional_sequence_lstm_test.cc | 3236 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
|
| D | lstm_test.cc | 1804 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 341 const TfLiteTensor* cell_to_forget_weights, in EvalFloat()
559 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 273 const TfLiteTensor* cell_to_forget_weights, in EvalCalibration()
515 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 49 std::initializer_list<float> cell_to_forget_weights, in LSTMOpModel()
|