Searched refs:cellStateOut (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | quantized_lstm.model.cpp | 26 auto cellStateOut = model->addOperand(&type4); in CreateModel() local
29 … forgetGateBias, cellGateBias, outputGateBias, prevCellState, prevOutput}, {cellStateOut, output}); in CreateModel()
33 {cellStateOut, output}); in CreateModel()
65 auto cellStateOut = model->addOperand(&type4); in CreateModel_relaxed() local
68 … forgetGateBias, cellGateBias, outputGateBias, prevCellState, prevOutput}, {cellStateOut, output}); in CreateModel_relaxed()
72 {cellStateOut, output}); in CreateModel_relaxed()
108 auto cellStateOut = model->addOperand(&type9); in CreateModel_dynamic_output_shape() local
111 … forgetGateBias, cellGateBias, outputGateBias, prevCellState, prevOutput}, {cellStateOut, output}); in CreateModel_dynamic_output_shape()
115 {cellStateOut, output}); in CreateModel_dynamic_output_shape()
149 auto cellStateOut = model->addOperand(&type9); in CreateModel_dynamic_output_shape_relaxed() local
[all …]
|
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 377 std::vector<float> cellStateOut(cellStateSize); in execute() local
412 outputStateOut.data(), cellStateOut.data(), in execute()
418 std::vector<_Float16> cellStateOut(cellStateSize); in execute() local
454 outputStateOut.data(), cellStateOut.data(), in execute()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1120 RunTimeOperandInfo& cellStateOut = mOperands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation() local
1130 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateShape, &result) && in executeOperation()
1683 RunTimeOperandInfo& cellStateOut = in executeOperation() local
1692 setInfoAndAllocateIfNeeded(&cellStateOut, cellStateOutShape, &result) && in executeOperation()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1724 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in TEST() local
1751 {scratch, outputStateOut, cellStateOut, output}); in TEST()
1812 ANeuralNetworksOperandType cellStateOut = floatTensor2D; in lstmTestV1_2() local
1843 {scratch, outputStateOut, cellStateOut, output}); in lstmTestV1_2()
|