Searched refs:forgetGateScratchDecoder (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | Lstm.cpp | 53 std::unique_ptr<Decoder<float>>& forgetGateScratchDecoder, in LstmImpl()
156 MeanStddevNormalization(*forgetGateScratchDecoder, in LstmImpl()
159 nCell, *forgetGateScratchDecoder, nBatch, *forgetGateScratch); in LstmImpl()
161 nCell, *forgetGateScratchDecoder, nBatch, *forgetGateScratch); in LstmImpl()
163 Activation(*forgetGateScratchDecoder, *forgetGateScratch, in LstmImpl()
178 … VectorVectorCwiseProduct(*forgetGateScratchDecoder, *cellStateIn, nBatch * nCell, *cellStateOut); in LstmImpl()
193 Sub1Vector(*forgetGateScratchDecoder, nBatch * nCell, *forgetGateScratch); in LstmImpl()
195 *cellScratchDecoder, *forgetGateScratchDecoder, nBatch * nCell, *cellStateOut); in LstmImpl()
|
| D | RefLstmWorkload.cpp | 91 std::unique_ptr<Decoder<float>> forgetGateScratchDecoder = in Execute() local
103 *forgetGateScratchDecoder += (1 * nCell * nBatch); in Execute()
115 *forgetGateScratchDecoder += (2 * nCell * nBatch); in Execute()
248 forgetGateScratchDecoder, in Execute()
|
| D | Lstm.hpp | 57 std::unique_ptr<Decoder<float>>& forgetGateScratchDecoder,
|
| D | RefUnidirectionalSequenceLstmWorkload.cpp | 108 std::unique_ptr<Decoder<float>> forgetGateScratchDecoder = MakeDecoder<float>(scratchInfo, in Execute() local
281 forgetGateScratchDecoder, in Execute()
|