Searched refs:recurrentToForgetWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 145 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local 159 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 173 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 335 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local 348 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TestImpl() 361 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl() 524 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local 537 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 550 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 838 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo6x5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local [all …]
|
H A D | LstmTestImpl.cpp | 275 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local 289 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmNoCifgNoPeepholeNoProjectionTestImpl() 303 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl() 934 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local 953 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 971 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 1119 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local 1134 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1149 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1394 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local [all …]
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | Lstm.cpp | 31 std::unique_ptr<Decoder<float>>& recurrentToForgetWeightsTensor, in LstmImpl() argument 119 MatrixBatchVectorMultiplyAccumulate(*recurrentToForgetWeightsTensor, in LstmImpl()
|
H A D | RefLstmWorkload.cpp | 128 std::unique_ptr<Decoder<float>> recurrentToForgetWeightsTensor = MakeDecoder<float>( in Execute() local 226 recurrentToForgetWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 159 std::unique_ptr<Decoder<float>> recurrentToForgetWeightsTensor = MakeDecoder<float>( in Execute() local 259 recurrentToForgetWeightsTensor, in Execute()
|
H A D | Lstm.hpp | 35 std::unique_ptr<Decoder<float>>& recurrentToForgetWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QLstmEndToEndTestImpl.cpp | 123 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local 153 params.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in QLstmEndToEnd()
|
H A D | QuantizedLstmEndToEndTestImpl.cpp | 80 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local 109 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 537 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3); 560 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor;
|