Searched refs:cellStateOutTensor (Results 1 – 1 of 1) sorted by relevance
387 Shape cellStateOutTensor = context->getInputShape(kCellStateInTensor); in prepare() local388 cellStateOutTensor.dimensions.resize(2); in prepare()389 cellStateOutTensor.dimensions[0] = batchSize; in prepare()390 cellStateOutTensor.dimensions[1] = numCells; in prepare()391 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()