Home
last modified time | relevance | path

Searched defs:recurrentToOutputWeightsShape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp17 const TensorShape& recurrentToOutputWeightsShape, in LstmImpl()
H A DRefLstmWorkload.cpp156 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp136 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQLSTM.cpp268 const Shape recurrentToOutputWeightsShape = in execute() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp1516 armnn::TensorShape recurrentToOutputWeightsShape = {4, 4}; variable