Searched refs:cellStateShape (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 215 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare() local
216 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2u); in prepare()
217 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
218 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numCells); in prepare()
|
| D | QLSTM.cpp | 186 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare() local
187 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2u); in prepare()
188 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
189 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numUnits); in prepare()
|
| D | LSTM.cpp | 304 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare() argument
385 cellStateShape->type = inputShape.type; in Prepare()
386 cellStateShape->dimensions = {n_batch, n_cell}; in Prepare()
387 cellStateShape->offset = inputShape.offset; in Prepare()
388 cellStateShape->scale = inputShape.scale; in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 34 Shape* cellStateShape, Shape* outputShape);
|
| D | LSTM.h | 53 Shape* outputStateShape, Shape* cellStateShape, Shape* outputShape);
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1047 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local
1051 &cellStateShape, &outputShape) && in executeOperation()
1054 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
|