Searched refs:stateShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDF.cpp | 51 bool SVDF::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape, in Prepare() argument
98 stateShape->type = inputShape.type; in Prepare()
99 stateShape->dimensions = {batch_size, memory_size * num_filters}; in Prepare()
100 stateShape->offset = inputShape.offset; in Prepare()
101 stateShape->scale = inputShape.scale; in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | SVDF.h | 43 static bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape,
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1089 Shape stateShape, outputShape; in executeOperation() local
1092 success = SVDF::Prepare(operation, operands, &stateShape, &outputShape) && in executeOperation()
1093 setInfoAndAllocateIfNeeded(&stateOut, stateShape, &result) && in executeOperation()
|