Home
last modified time | relevance | path

Searched refs:ExtractQuantLstmWeightsSubmatrix (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup.cc27 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims, in ExtractQuantLstmWeightsSubmatrix() function
98 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 0 * output_depth, in DecomposeQuantLstmWeightsTensor()
101 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 0 * output_depth, in DecomposeQuantLstmWeightsTensor()
105 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 1 * output_depth, in DecomposeQuantLstmWeightsTensor()
108 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 1 * output_depth, in DecomposeQuantLstmWeightsTensor()
112 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 2 * output_depth, in DecomposeQuantLstmWeightsTensor()
115 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 2 * output_depth, in DecomposeQuantLstmWeightsTensor()
119 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 3 * output_depth, in DecomposeQuantLstmWeightsTensor()
122 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 3 * output_depth, in DecomposeQuantLstmWeightsTensor()
Dquant_lstm_sup_test.cc59 using tflite::delegate::nnapi::ExtractQuantLstmWeightsSubmatrix;
77 ExtractQuantLstmWeightsSubmatrix(submatrix_dims, 0 /* offset_row */, in TEST_F()
98 ExtractQuantLstmWeightsSubmatrix(submatrix_dims, 0 /* offset_row */, in TEST_F()
119 ExtractQuantLstmWeightsSubmatrix( in TEST_F()
Dquant_lstm_sup.h26 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims,