Searched refs:cellStateOut (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 414 float* cellStateOut; in execute() local
417 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()
422 cellStateOut = cellStateOutBuffer.data(); in execute()
458 cellStateOut, context->getOutputBuffer<float>(kOutputTensor), in execute()
466 _Float16* cellStateOut; in execute() local
469 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
474 cellStateOut = cellStateOutBuffer.data(); in execute()
511 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1043 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation() local
1053 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
1619 RunTimeOperandInfo& cellStateOut = in executeOperation() local
1628 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateOutShape, &result) && in executeOperation()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2645 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in TEST() local
2672 {scratch, outputStateOut, cellStateOut, output}); in TEST()
2733 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in lstmTestV1_2() local
2764 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|