Searched refs:outputStateShape (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceRNN.cpp | 155 Shape outputStateShape = context->getInputShape(kHiddenStateTensor); in prepare() local
156 outputStateShape.dimensions.resize(2); in prepare()
157 outputStateShape.dimensions[0] = batchSize; in prepare()
158 outputStateShape.dimensions[1] = numUnits; in prepare()
159 NN_RET_CHECK(context->setOutputShape(kStateOutputTensor, outputStateShape)); in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 211 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare() local
212 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2u); in prepare()
213 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 1), outputSize); in prepare()
|
| D | QLSTM.cpp | 182 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare() local
183 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2u); in prepare()
184 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
185 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 1), outputSize); in prepare()
|
| D | LSTM.cpp | 304 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare() argument
380 outputStateShape->type = inputShape.type; in Prepare()
381 outputStateShape->dimensions = {n_batch, n_output}; in Prepare()
382 outputStateShape->offset = inputShape.offset; in Prepare()
383 outputStateShape->scale = inputShape.scale; in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| 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
1050 success = lstm_cell.Prepare(operation, operands, &scratchShape, &outputStateShape, in executeOperation()
1053 setInfoAndAllocateIfNeeded(&outputStateOut, outputStateShape, &result) && in executeOperation()
|