Searched refs:inputToForgetWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 141 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 D | LstmTestImpl.cpp | 271 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 D | Lstm.cpp | 27 std::unique_ptr<Decoder<float>>& inputToForgetWeightsTensor, in LstmImpl() argument 106 MatrixBatchVectorMultiplyAccumulate(*inputToForgetWeightsTensor, in LstmImpl()
|
H A D | RefLstmWorkload.cpp | 120 std::unique_ptr<Decoder<float>> inputToForgetWeightsTensor = MakeDecoder<float>( in Execute() local 222 inputToForgetWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 151 std::unique_ptr<Decoder<float>> inputToForgetWeightsTensor = MakeDecoder<float>( in Execute() local 255 inputToForgetWeightsTensor, in Execute()
|
H A D | Lstm.hpp | 31 std::unique_ptr<Decoder<float>>& inputToForgetWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QLstmEndToEndTestImpl.cpp | 112 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in QLstmEndToEnd() local 149 params.m_InputToForgetWeights = &inputToForgetWeightsTensor; in QLstmEndToEnd()
|
H A D | QuantizedLstmEndToEndTestImpl.cpp | 66 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in CreateQuantizedLstmNetwork() local 105 data.m_InputToForgetWeights = &inputToForgetWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 534 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo4x5); 556 data.m_InputToForgetWeights = &inputToForgetWeightsTensor;
|