Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp148 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
842 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo6); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
1068 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1745 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1938 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfoNum); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
H A DLstmTestImpl.cpp278 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
938 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo20); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1398 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp34 std::unique_ptr<Decoder<float>>& cellToInputWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp143 std::unique_ptr<Decoder<float>> cellToInputWeightsTensor; in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp174 std::unique_ptr<Decoder<float>> cellToInputWeightsTensor; in Execute() local