Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp16 const TensorShape& inputToOutputWeightsShape, in LstmImpl()
H A DRefLstmWorkload.cpp155 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp135 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQLSTM.cpp267 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp1479 armnn::TensorShape inputToOutputWeightsShape = {4, 2}; variable