Searched refs:recurrent_submatrix_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup.cc | 56 TfLiteIntArray* recurrent_submatrix_dims, in SetWeightSubmatrixDims() argument
61 recurrent_submatrix_dims->data[0] = output_depth; in SetWeightSubmatrixDims()
62 recurrent_submatrix_dims->data[1] = output_depth; in SetWeightSubmatrixDims()
93 TfLiteIntArray* recurrent_submatrix_dims = TfLiteIntArrayCreate(2); in DecomposeQuantLstmWeightsTensor() local
95 SetWeightSubmatrixDims(weight_dims, recurrent_submatrix_dims, in DecomposeQuantLstmWeightsTensor()
98 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 0 * output_depth, in DecomposeQuantLstmWeightsTensor()
105 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 1 * output_depth, in DecomposeQuantLstmWeightsTensor()
112 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 2 * output_depth, in DecomposeQuantLstmWeightsTensor()
119 ExtractQuantLstmWeightsSubmatrix(recurrent_submatrix_dims, 3 * output_depth, in DecomposeQuantLstmWeightsTensor()
126 TfLiteIntArrayFree(recurrent_submatrix_dims); in DecomposeQuantLstmWeightsTensor()
|
| D | quant_lstm_sup.h | 45 TfLiteIntArray* recurrent_submatrix_dims,
|