Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup_test.cc75 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 3}); in TEST_F() local
77 ExtractQuantLstmWeightsSubmatrix(submatrix_dims, 0 /* offset_row */, in TEST_F()
96 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 2}); in TEST_F() local
98 ExtractQuantLstmWeightsSubmatrix(submatrix_dims, 0 /* offset_row */, in TEST_F()
117 const TfLiteIntArray* submatrix_dims = CreateDimArray(2, {2, 2}); in TEST_F() local
120 submatrix_dims, 1 * submatrix_dims->data[0] /* offset_row */, in TEST_F()
Dquant_lstm_sup.cc27 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims, in ExtractQuantLstmWeightsSubmatrix() argument
33 auto const& submatrix_rows = submatrix_dims->data[0]; in ExtractQuantLstmWeightsSubmatrix()
34 auto const& submatrix_cols = submatrix_dims->data[1]; in ExtractQuantLstmWeightsSubmatrix()
37 submatrix->resize(NumElements(submatrix_dims)); in ExtractQuantLstmWeightsSubmatrix()
Dquant_lstm_sup.h26 void ExtractQuantLstmWeightsSubmatrix(const TfLiteIntArray* submatrix_dims,