Home
last modified time | relevance | path

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

/external/armnn/src/backends/reference/workloads/
DLstm.cpp16 const TensorShape& inputToOutputWeightsShape, in LstmImpl()
DRefLstmWorkload.cpp155 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute() local
DRefUnidirectionalSequenceLstmWorkload.cpp135 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute() local
/external/armnn/src/armnnSerializer/test/
DLstmSerializationTests.cpp1479 armnn::TensorShape inputToOutputWeightsShape = {4, 2}; variable