Searched refs:hiddenStateShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 45 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare() argument
71 hiddenStateShape->type = inputShape.type; in Prepare()
72 hiddenStateShape->dimensions = {batch_size, num_units}; in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 42 Shape* hiddenStateShape, Shape* outputShape);
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1078 Shape hiddenStateShape, outputShape; in executeOperation() local
1081 success = RNN::Prepare(operation, operands, &hiddenStateShape, &outputShape) && in executeOperation()
1082 setInfoAndAllocateIfNeeded(&hiddenStateOut, hiddenStateShape, &result) && in executeOperation()
|