Searched defs:cellStateShape (Results 1 – 3 of 3) sorted by relevance
308 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare() local
295 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare()
1123 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local