Searched refs:cellStateOutShape (Results 1 – 2 of 2) sorted by relevance
243 std::vector<RunTimeOperandInfo>& operands, Shape* cellStateOutShape, in prepare() argument330 *cellStateOutShape = prevCellState->shape(); in prepare()
1687 Shape cellStateOutShape, outputShape; in executeOperation() local1690 success = QuantizedLSTMCell::prepare(operation, mOperands, &cellStateOutShape, in executeOperation()1692 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateOutShape, &result) && in executeOperation()