Home
last modified time | relevance | path

Searched defs:forget_layer_norm_coefficients (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc135 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
626 const TfLiteTensor* forget_layer_norm_coefficients; in CheckInputTensorDimensions() local
855 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in Prepare() local
1290 const TfLiteTensor* forget_layer_norm_coefficients = in Eval() local
Dlstm.cc220 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
508 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
1078 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1296 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in Prepare() local
1871 const TfLiteTensor* forget_layer_norm_coefficients = in Eval() local
Dlstm_test.cc1641 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
Dlstm_eval.cc1700 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat()
1889 const TfLiteTensor* forget_layer_norm_coefficients, in EvalHybrid()
2163 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_16()
2401 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_8()
Dunidirectional_sequence_lstm_test.cc3077 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
3241 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc344 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat()
567 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc276 const TfLiteTensor* forget_layer_norm_coefficients, in EvalCalibration()
523 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc58 std::initializer_list<float> forget_layer_norm_coefficients, in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2048 const auto forget_layer_norm_coefficients = in Verify() local