Searched refs:scratchShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 304 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare() argument
392 scratchShape->dimensions = {n_batch, n_cell * 3}; in Prepare()
395 scratchShape->dimensions = {n_batch, n_cell * 4}; in Prepare()
397 scratchShape->type = inputShape.type; in Prepare()
398 scratchShape->offset = inputShape.offset; in Prepare()
399 scratchShape->scale = inputShape.scale; in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 52 bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* scratchShape,
|
| /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()
1052 setInfoAndAllocateIfNeeded(&scratch, scratchShape, &result) && in executeOperation()
|