Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp440 Shape batchInputShape = input_shape; in LSTMEvalFloat32() local
560 Shape batchInputShape = input_shape; in LSTMEvalFloat16() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefUnidirectionalSequenceLstmWorkload.cpp129 TensorShape batchInputShape = TensorShape({batchSize, inputSize}); in Execute() local