Searched refs:outputStateOutTensor (Results 1 – 1 of 1) sorted by relevance
384 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare() local385 outputStateOutTensor.dimensions.resize(2); in prepare()386 outputStateOutTensor.dimensions[0] = batchSize; in prepare()387 outputStateOutTensor.dimensions[1] = outputSize; in prepare()388 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()