Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp141 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
155 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
169 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
331 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
344 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
357 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
520 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
533 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
546 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
835 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
H A DLstmTestImpl.cpp271 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
285 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
299 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
931 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
949 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
967 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1111 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfoInput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1126 AllocateAndCopyDataToITensorHandle(&inputToForgetWeightsTensor, inputToForgetWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1141 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1391 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp27 std::unique_ptr<Decoder<float>>& inputToForgetWeightsTensor, in LstmImpl() argument
106 MatrixBatchVectorMultiplyAccumulate(*inputToForgetWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp120 std::unique_ptr<Decoder<float>> inputToForgetWeightsTensor = MakeDecoder<float>( in Execute() local
222 inputToForgetWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp151 std::unique_ptr<Decoder<float>> inputToForgetWeightsTensor = MakeDecoder<float>( in Execute() local
255 inputToForgetWeightsTensor, in Execute()
H A DLstm.hpp31 std::unique_ptr<Decoder<float>>& inputToForgetWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp112 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in QLstmEndToEnd() local
149 params.m_InputToForgetWeights = &inputToForgetWeightsTensor; in QLstmEndToEnd()
H A DQuantizedLstmEndToEndTestImpl.cpp66 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in CreateQuantizedLstmNetwork() local
105 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp534 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo4x5);
556 data.m_InputToForgetWeights = &inputToForgetWeightsTensor;