Searched refs:recurrentToOutputWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 147 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local 161 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 175 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 337 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local 350 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TestImpl() 363 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl() 526 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local 539 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 552 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 841 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo6x5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local [all …]
|
H A D | LstmTestImpl.cpp | 277 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local 291 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in LstmNoCifgNoPeepholeNoProjectionTestImpl() 305 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl() 937 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local 955 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 973 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 1120 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local 1135 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, recurrentToOutputWeights.data(… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1150 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1397 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local [all …]
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | Lstm.cpp | 33 std::unique_ptr<Decoder<float>>& recurrentToOutputWeightsTensor, in LstmImpl() argument 123 MatrixBatchVectorMultiplyAccumulate(*recurrentToOutputWeightsTensor, in LstmImpl()
|
H A D | RefLstmWorkload.cpp | 132 std::unique_ptr<Decoder<float>> recurrentToOutputWeightsTensor = MakeDecoder<float>( in Execute() local 228 recurrentToOutputWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 163 std::unique_ptr<Decoder<float>> recurrentToOutputWeightsTensor = MakeDecoder<float>( in Execute() local 261 recurrentToOutputWeightsTensor, in Execute()
|
H A D | Lstm.hpp | 37 std::unique_ptr<Decoder<float>>& recurrentToOutputWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QLstmEndToEndTestImpl.cpp | 127 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local 155 params.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in QLstmEndToEnd()
|
H A D | QuantizedLstmEndToEndTestImpl.cpp | 89 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local 111 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 540 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3); 562 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor;
|