recurrentToOutputWeightsShape (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A D	RefUnidirectionalSequenceLstmWorkload.cpp	`136 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute() local 137 unsigned int nOutput = recurrentToOutputWeightsShape[1]; in Execute() 245 recurrentToOutputWeightsShape, in Execute()`
H A D	Lstm.cpp	`17 const TensorShape& recurrentToOutputWeightsShape, in LstmImpl() argument 67 const uint32_t nOutput = recurrentToOutputWeightsShape[1]; in LstmImpl()`
H A D	RefLstmWorkload.cpp	`156 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute() local 212 recurrentToOutputWeightsShape, in Execute()`
H A D	Lstm.hpp	`21 const TensorShape& recurrentToOutputWeightsShape,`
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
D	QLSTM.cpp	`268 const Shape recurrentToOutputWeightsShape = in execute() local 279 const uint32_t outputSize = recurrentToOutputWeightsShape.dimensions[1]; in execute() 444 recurrentToOutputWeightsShape.scale * prevOutputShape.scale / outputIntermediateScale; in execute() 530 -prevOutputShape.offset, recurrentToOutputWeightsBuffer, recurrentToOutputWeightsShape, in execute()`
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A D	LstmSerializationTests.cpp	`1516 armnn::TensorShape recurrentToOutputWeightsShape = {4, 4}; variable 1518 armnn::TensorInfo recurrentToOutputWeightsInfo(recurrentToOutputWeightsShape,`