Searched refs:outputStateOut (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 410 float* outputStateOut; in execute() local
413 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
418 outputStateOut = outputStateOutBuffer.data(); in execute()
454 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), outputStateOut, in execute()
462 _Float16* outputStateOut; in execute() local
465 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
470 outputStateOut = outputStateOutBuffer.data(); in execute()
508 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2645 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in TEST() local
2673 {scratch, outputStateOut, cellStateOut, output}); in TEST()
2733 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in lstmTestV1_2() local
2765 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1098 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation() local
1108 setInfoAndAllocateIfNeeded(&outputStateOut, outputStateShape, &result) && in executeOperation()
|