Searched refs:output_state_data_int16 (Results 1 – 1 of 1) sorted by relevance
64 const Dims<4>& prevCellState_dims, int16_t* output_state_data_int16, in quantizedLstmStep() argument194 output_state_data_int16[b * output_depth + c] = new_state.raw(); in quantizedLstmStep()