Searched refs:prevCellState (Results 1 – 1 of 1) sorted by relevance
174 FS prevCellState = FS::FromRaw(prevCellState_data_int16[b * output_depth + c]); in quantizedLstmStep() local175 FS prevCellState_times_forget_state = forget_gate_output * prevCellState; in quantizedLstmStep()317 auto prevCellState = GetInput(operation, operands, kPrevCellStateTensor); in prepare() local318 NN_CHECK_EQ(NumDimensions(prevCellState), 2); in prepare()319 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()320 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()321 NN_CHECK_EQ(prevCellState->zeroPoint, 0); in prepare()330 NN_CHECK(tflite::CheckedLog2(prevCellState->scale, &stateScaleLog2Rounded)); in prepare()335 *cellStateOutShape = prevCellState->shape(); in prepare()