Home
last modified time | relevance | path

Searched refs:recurrentToInputWeightsTensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp144 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
158 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
172 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
334 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
347 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
360 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
523 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
536 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
549 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
839 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo6x5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
H A DLstmTestImpl.cpp274 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
288 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
302 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
935 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
952 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
970 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1395 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1417 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, recurrentToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1440 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1613 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(recurrentWeightsInfo); in QuantizedLstmTestImpl() local
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp127 std::unique_ptr<Decoder<float>> recurrentToInputWeightsTensor; in Execute() local
179 recurrentToInputWeightsTensor = MakeDecoder<float>( in Execute()
225 recurrentToInputWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp158 std::unique_ptr<Decoder<float>> recurrentToInputWeightsTensor; in Execute() local
207 recurrentToInputWeightsTensor = MakeDecoder<float>( in Execute()
258 recurrentToInputWeightsTensor, in Execute()
H A DLstm.cpp30 std::unique_ptr<Decoder<float>>& recurrentToInputWeightsTensor, in LstmImpl() argument
116 MatrixBatchVectorMultiplyAccumulate(*recurrentToInputWeightsTensor, in LstmImpl()
H A DLstm.hpp34 std::unique_ptr<Decoder<float>>& recurrentToInputWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQuantizedLstmEndToEndTestImpl.cpp76 …armnn::ConstTensor recurrentToInputWeightsTensor(recurrentWeightsInfo, recurrentToInputWeightsTens… in CreateQuantizedLstmNetwork() local
108 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp538 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3);
559 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor;