Searched refs:inputToInputWeightsTensor (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 140 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local 154 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 168 data.m_InputToInputWeights = &inputToInputWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 330 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local 343 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl() 356 data.m_InputToInputWeights = &inputToInputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl() 519 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local 532 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 545 data.m_InputToInputWeights = &inputToInputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 834 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local [all …]
|
H A D | LstmTestImpl.cpp | 270 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local 284 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl() 298 data.m_InputToInputWeights = &inputToInputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl() 930 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local 948 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 966 data.m_InputToInputWeights = &inputToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 1390 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local 1413 AllocateAndCopyDataToITensorHandle(&inputToInputWeightsTensor, inputToInputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() 1436 data.m_InputToInputWeights = &inputToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() 1608 armnn::ScopedTensorHandle inputToInputWeightsTensor(inputWeightsInfo); in QuantizedLstmTestImpl() local [all …]
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | RefLstmWorkload.cpp | 119 std::unique_ptr<Decoder<float>> inputToInputWeightsTensor; in Execute() local 175 inputToInputWeightsTensor = MakeDecoder<float>( in Execute() 221 inputToInputWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 150 std::unique_ptr<Decoder<float>> inputToInputWeightsTensor; in Execute() local 203 inputToInputWeightsTensor = MakeDecoder<float>( in Execute() 254 inputToInputWeightsTensor, in Execute()
|
H A D | Lstm.cpp | 26 std::unique_ptr<Decoder<float>>& inputToInputWeightsTensor, in LstmImpl() argument 103 MatrixBatchVectorMultiplyAccumulate(*inputToInputWeightsTensor, in LstmImpl()
|
H A D | Lstm.hpp | 30 std::unique_ptr<Decoder<float>>& inputToInputWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QuantizedLstmEndToEndTestImpl.cpp | 63 … armnn::ConstTensor inputToInputWeightsTensor(inputWeightsInfo, inputToInputWeightsVector.data()); in CreateQuantizedLstmNetwork() local 104 data.m_InputToInputWeights = &inputToInputWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 533 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo4x5); 555 data.m_InputToInputWeights = &inputToInputWeightsTensor;
|