Searched refs:inputGateScratchDecoder (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | Lstm.cpp | 51 std::unique_ptr<Decoder<float>>& inputGateScratchDecoder, in LstmImpl()
136 MeanStddevNormalization(*inputGateScratchDecoder, in LstmImpl()
139 … nCell, *inputGateScratchDecoder, nBatch, *inputGateScratch); in LstmImpl()
141 nCell, *inputGateScratchDecoder, nBatch, *inputGateScratch); in LstmImpl()
143 Activation(*inputGateScratchDecoder, *inputGateScratch, in LstmImpl()
200 *cellScratchDecoder, *inputGateScratchDecoder, nBatch * nCell, *cellStateOut); in LstmImpl()
|
| D | RefLstmWorkload.cpp | 87 std::unique_ptr<Decoder<float>> inputGateScratchDecoder = in Execute() local
113 *inputGateScratchDecoder += (0 * nCell * nBatch); in Execute()
246 inputGateScratchDecoder, in Execute()
|
| D | RefUnidirectionalSequenceLstmWorkload.cpp | 106 std::unique_ptr<Decoder<float>> inputGateScratchDecoder; in Execute() local
121 inputGateScratchDecoder = MakeDecoder<float>(scratchInfo, inputGateScratchBuffer.data()); in Execute()
279 inputGateScratchDecoder, in Execute()
|
| D | Lstm.hpp | 55 std::unique_ptr<Decoder<float>>& inputGateScratchDecoder,
|