Searched refs:output_depth (Results 1 – 1 of 1) sorted by relevance
84 const int output_depth = in quantizedLstmStep() local87 TFLITE_CHECK_EQ(output_depth, intern_activ_depth / 4); in quantizedLstmStep()92 TFLITE_CHECK_EQ(fc_output_depth, 4 * output_depth); in quantizedLstmStep()134 for (int c = 0; c < output_depth; ++c) { 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()174 FS prevCellState = FS::FromRaw(prevCellState_data_int16[b * output_depth + c]); in quantizedLstmStep()193 output_state_data_int16[b * output_depth + c] = new_state.raw(); in quantizedLstmStep()[all …]