Searched refs:outputStateOut (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 376 std::vector<float> outputStateOut(outputStateSize); in execute() local
412 outputStateOut.data(), cellStateOut.data(), in execute()
417 std::vector<_Float16> outputStateOut(outputStateSize); in execute() local
454 outputStateOut.data(), cellStateOut.data(), in execute()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1723 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in TEST() local
1751 {scratch, outputStateOut, cellStateOut, output}); in TEST()
1811 ANeuralNetworksOperandType outputStateOut = floatTensor2D; in lstmTestV1_2() local
1843 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1119 RunTimeOperandInfo& outputStateOut = mOperands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation() local
1129 setInfoAndAllocateIfNeeded(&outputStateOut, outputStateShape, &result) && in executeOperation()
|