Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp847 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfo6); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
1073 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfo5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1274 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfo4); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1750 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1943 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
2120 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest() local
H A DLstmTestImpl.cpp943 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfo20); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1122 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfoNumUnits); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1403 armnn::ScopedTensorHandle cellToForgetWeightsTensor(tensorInfo4); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp35 std::unique_ptr<Decoder<float>>& cellToForgetWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp144 std::unique_ptr<Decoder<float>> cellToForgetWeightsTensor; in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp175 std::unique_ptr<Decoder<float>> cellToForgetWeightsTensor; in Execute() local