Searched defs:forget_layer_norm_coefficients (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_lstm.cc | 140 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
634 const TfLiteTensor* forget_layer_norm_coefficients; in CheckInputTensorDimensions() local
865 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in Prepare() local
1303 const TfLiteTensor* forget_layer_norm_coefficients = in Eval() local
|
| H A D | lstm.cc | 221 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
509 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
1079 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1297 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in Prepare() local
1875 const TfLiteTensor* forget_layer_norm_coefficients = in Eval() local
|
| H A D | lstm_test.cc | 1641 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
1809 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
1976 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
|
| H A D | lstm_eval.cc | 1758 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat()
1953 const TfLiteTensor* forget_layer_norm_coefficients, in EvalHybrid()
2227 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_16()
2465 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_8()
|
| H A D | unidirectional_sequence_lstm_test.cc | 2825 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
2989 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 353 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat()
579 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 277 const TfLiteTensor* forget_layer_norm_coefficients, in EvalCalibration()
524 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | lstm_full_test.cc | 58 std::initializer_list<float> forget_layer_norm_coefficients, in LSTMOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 2541 const auto forget_layer_norm_coefficients = in verify() local
|