Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.cpp89 const int fc_output_depth = in quantizedLstmStep() local
92 TFLITE_CHECK_EQ(fc_output_depth, 4 * output_depth); in quantizedLstmStep()
108 for (int out_c = 0; out_c < fc_output_depth; ++out_c) { in quantizedLstmStep()
127 activ_temp_data_int16[out_c + fc_output_depth * b] = accum; in quantizedLstmStep()
155 activ_temp_data_int16[b * fc_output_depth + 0 * output_depth + c]); in quantizedLstmStep()
160 activ_temp_data_int16[b * fc_output_depth + 1 * output_depth + c]); in quantizedLstmStep()
165 activ_temp_data_int16[b * fc_output_depth + 2 * output_depth + c]); in quantizedLstmStep()
169 activ_temp_data_int16[b * fc_output_depth + 3 * output_depth + c]); in quantizedLstmStep()