Home
last modified time | relevance | path

Searched refs:batchInputShape (Results 1 – 1 of 1) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DRefUnidirectionalSequenceLstmWorkload.cpp129 TensorShape batchInputShape = TensorShape({batchSize, inputSize}); in Execute() local
130 lstmInputInfo.SetShape(batchInputShape); in Execute()