Searched refs:recurrentToInputWeightsTensor (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 144 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 D | LstmTestImpl.cpp | 274 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 D | RefLstmWorkload.cpp | 127 std::unique_ptr<Decoder<float>> recurrentToInputWeightsTensor; in Execute() local 179 recurrentToInputWeightsTensor = MakeDecoder<float>( in Execute() 225 recurrentToInputWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 158 std::unique_ptr<Decoder<float>> recurrentToInputWeightsTensor; in Execute() local 207 recurrentToInputWeightsTensor = MakeDecoder<float>( in Execute() 258 recurrentToInputWeightsTensor, in Execute()
|
H A D | Lstm.cpp | 30 std::unique_ptr<Decoder<float>>& recurrentToInputWeightsTensor, in LstmImpl() argument 116 MatrixBatchVectorMultiplyAccumulate(*recurrentToInputWeightsTensor, in LstmImpl()
|
H A D | Lstm.hpp | 34 std::unique_ptr<Decoder<float>>& recurrentToInputWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QuantizedLstmEndToEndTestImpl.cpp | 76 …armnn::ConstTensor recurrentToInputWeightsTensor(recurrentWeightsInfo, recurrentToInputWeightsTens… in CreateQuantizedLstmNetwork() local 108 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 538 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3); 559 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor;
|