Searched refs:cellStateSize (Results 1 – 1 of 1) sorted by relevance
399 const auto cellStateSize = getNumberOfElements(context->getInputShape(kCellStateInTensor)); in execute() local401 const auto scratchSize = use_cifg ? 3 * cellStateSize : 4 * cellStateSize; in execute()417 cellStateOutBuffer.resize(cellStateSize); in execute()469 cellStateOutBuffer.resize(cellStateSize); in execute()