Home
last modified time | relevance | path

Searched refs:inputToInputWeightsTensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp140 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 DLstmTestImpl.cpp270 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 DRefLstmWorkload.cpp119 std::unique_ptr<Decoder<float>> inputToInputWeightsTensor; in Execute() local
175 inputToInputWeightsTensor = MakeDecoder<float>( in Execute()
221 inputToInputWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp150 std::unique_ptr<Decoder<float>> inputToInputWeightsTensor; in Execute() local
203 inputToInputWeightsTensor = MakeDecoder<float>( in Execute()
254 inputToInputWeightsTensor, in Execute()
H A DLstm.cpp26 std::unique_ptr<Decoder<float>>& inputToInputWeightsTensor, in LstmImpl() argument
103 MatrixBatchVectorMultiplyAccumulate(*inputToInputWeightsTensor, in LstmImpl()
H A DLstm.hpp30 std::unique_ptr<Decoder<float>>& inputToInputWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQuantizedLstmEndToEndTestImpl.cpp63 … armnn::ConstTensor inputToInputWeightsTensor(inputWeightsInfo, inputToInputWeightsVector.data()); in CreateQuantizedLstmNetwork() local
104 data.m_InputToInputWeights = &inputToInputWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp533 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo4x5);
555 data.m_InputToInputWeights = &inputToInputWeightsTensor;