Searched refs:input_submatrix_dims (Results 1 – 2 of 2) sorted by relevance
57 TfLiteIntArray* input_submatrix_dims) { in SetWeightSubmatrixDims() argument64 input_submatrix_dims->data[0] = output_depth; in SetWeightSubmatrixDims()65 input_submatrix_dims->data[1] = input_depth; in SetWeightSubmatrixDims()94 TfLiteIntArray* input_submatrix_dims = TfLiteIntArrayCreate(2); in DecomposeQuantLstmWeightsTensor() local96 input_submatrix_dims); in DecomposeQuantLstmWeightsTensor()101 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 0 * output_depth, in DecomposeQuantLstmWeightsTensor()108 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 1 * output_depth, in DecomposeQuantLstmWeightsTensor()115 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 2 * output_depth, in DecomposeQuantLstmWeightsTensor()122 ExtractQuantLstmWeightsSubmatrix(input_submatrix_dims, 3 * output_depth, in DecomposeQuantLstmWeightsTensor()127 TfLiteIntArrayFree(input_submatrix_dims); in DecomposeQuantLstmWeightsTensor()
46 TfLiteIntArray* input_submatrix_dims);