Searched refs:cellStateShape (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.h | 34 Shape* cellStateShape, Shape* outputShape);
|
| D | UnidirectionalSequenceLSTM.cpp | 321 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare() local
322 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2); in prepare()
323 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
324 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numCells); in prepare()
|
| D | QLSTM.cpp | 301 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare() local
302 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2); in prepare()
303 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
304 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()
|
| D | LSTM.h | 54 Shape* outputStateShape, Shape* cellStateShape, Shape* outputShape);
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1100 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local
1104 &cellStateShape, &outputShape) && in executeOperation()
1107 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
|