Searched refs:outputStateOut (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 304 float* outputStateOut; in execute() local
307 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
312 outputStateOut = outputStateOutBuffer.data(); in execute()
348 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), outputStateOut, in execute()
356 _Float16* outputStateOut; in execute() local
359 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
364 outputStateOut = outputStateOutBuffer.data(); in execute()
402 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1043 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation() local
1053 setInfoAndAllocateIfNeeded(&outputStateOut, outputStateShape, &result) && in executeOperation()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2683 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in TEST() local
2711 {scratch, outputStateOut, cellStateOut, output}); in TEST()
2771 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in lstmTestV1_2() local
2803 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|