Home
last modified time | relevance | path

Searched defs:batchOutputSize (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp443 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat32() local
563 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat16() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefUnidirectionalSequenceLstmWorkload.cpp237 unsigned int batchOutputSize = batchSize * nOutput; in Execute() local