Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQuantizedLSTM.cpp86 const int fc_output_depth = in quantizedLstmStep() local
89 TFLITE_CHECK_EQ(fc_output_depth, 4 * output_depth); in quantizedLstmStep()
105 for (int out_c = 0; out_c < fc_output_depth; ++out_c) { in quantizedLstmStep()
124 activ_temp_data_int16[out_c + fc_output_depth * b] = accum; in quantizedLstmStep()
152 activ_temp_data_int16[b * fc_output_depth + 0 * output_depth + c]); in quantizedLstmStep()
157 activ_temp_data_int16[b * fc_output_depth + 1 * output_depth + c]); in quantizedLstmStep()
162 activ_temp_data_int16[b * fc_output_depth + 2 * output_depth + c]); in quantizedLstmStep()
166 activ_temp_data_int16[b * fc_output_depth + 3 * output_depth + c]); in quantizedLstmStep()