Home
last modified time | relevance | path

Searched refs:recurrentToOutputWeightsTensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp147 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 DLstmTestImpl.cpp277 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 DLstm.cpp33 std::unique_ptr<Decoder<float>>& recurrentToOutputWeightsTensor, in LstmImpl() argument
123 MatrixBatchVectorMultiplyAccumulate(*recurrentToOutputWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp132 std::unique_ptr<Decoder<float>> recurrentToOutputWeightsTensor = MakeDecoder<float>( in Execute() local
228 recurrentToOutputWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp163 std::unique_ptr<Decoder<float>> recurrentToOutputWeightsTensor = MakeDecoder<float>( in Execute() local
261 recurrentToOutputWeightsTensor, in Execute()
H A DLstm.hpp37 std::unique_ptr<Decoder<float>>& recurrentToOutputWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp127 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local
155 params.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in QLstmEndToEnd()
H A DQuantizedLstmEndToEndTestImpl.cpp89 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local
111 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp540 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3);
562 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor;