Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp442 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat32() local
562 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat16() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefUnidirectionalSequenceLstmWorkload.cpp236 unsigned int batchInputSize = batchSize * inputSize; in Execute() local