Searched refs:cellStateOut (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 305 float* cellStateOut; in execute() local
308 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()
313 cellStateOut = cellStateOutBuffer.data(); in execute()
349 cellStateOut, context->getOutputBuffer<float>(kOutputTensor), in execute()
357 _Float16* cellStateOut; in execute() local
360 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
365 cellStateOut = cellStateOutBuffer.data(); in execute()
402 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation() local
1054 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
1620 RunTimeOperandInfo& cellStateOut = in executeOperation() local
1629 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateOutShape, &result) && in executeOperation()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2684 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in TEST() local
2711 {scratch, outputStateOut, cellStateOut, output}); in TEST()
2772 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in lstmTestV1_2() local
2803 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|