Home
last modified time | relevance | path

Searched defs:cellStateShape (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp215 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare() local
DQLSTM.cpp186 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare() local
DLSTM.cpp304 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1047 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local