Searched defs:cell_layer_norm_coefficients (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | unidirectional_sequence_lstm.cc | 142 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local 650 const TfLiteTensor* cell_layer_norm_coefficients; in CheckInputTensorDimensions() local 1307 const TfLiteTensor* cell_layer_norm_coefficients = in Eval() local
|
H A D | lstm.cc | 223 const TfLiteTensor* cell_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local 511 const TfLiteTensor* cell_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local 1093 const TfLiteTensor* cell_layer_norm_coefficients = in CheckInputTensorDimensions() local 1877 const TfLiteTensor* cell_layer_norm_coefficients = in Eval() local
|
H A D | lstm_test.cc | 1643 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local 1811 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local 1978 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
|
H A D | lstm_eval.cc | 1759 const TfLiteTensor* cell_layer_norm_coefficients, in EvalFloat() 1954 const TfLiteTensor* cell_layer_norm_coefficients, in EvalHybrid() 2228 const TfLiteTensor* cell_layer_norm_coefficients, in EvalInteger8x8_16() 2466 const TfLiteTensor* cell_layer_norm_coefficients, in EvalInteger8x8_8()
|
H A D | unidirectional_sequence_lstm_test.cc | 2827 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local 2991 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
H A D | lstm.cc | 354 const TfLiteTensor* cell_layer_norm_coefficients, in EvalFloat() 582 const TfLiteTensor* cell_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 | 278 const TfLiteTensor* cell_layer_norm_coefficients, in EvalCalibration() 527 const TfLiteTensor* cell_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 | 35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel()
|