Searched refs:cellStateShape (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.h | 20 std::vector<RunTimeOperandInfo>& operands, Shape* cellStateShape,
|
| D | UnidirectionalSequenceLSTM.cpp | 308 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare() local
309 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2); in prepare()
310 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
311 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numCells); in prepare()
|
| D | LSTM.cpp | 295 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare() argument
342 cellStateShape->type = inputShape.type; in Prepare()
343 cellStateShape->dimensions = {n_batch, n_cell}; in Prepare()
344 cellStateShape->offset = inputShape.offset; in Prepare()
345 cellStateShape->scale = inputShape.scale; in Prepare()
|
| D | LSTM.h | 51 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1123 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local
1127 &cellStateShape, &outputShape) && in executeOperation()
1130 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
|