Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp143 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
157 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
171 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
333 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
346 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
359 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
522 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
535 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
548 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
837 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
H A DLstmTestImpl.cpp273 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
287 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
301 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
933 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
951 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
969 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1112 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfoInput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1127 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1142 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1393 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp29 std::unique_ptr<Decoder<float>>& inputToOutputWeightsTensor, in LstmImpl() argument
110 MatrixBatchVectorMultiplyAccumulate(*inputToOutputWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp124 std::unique_ptr<Decoder<float>> inputToOutputWeightsTensor = MakeDecoder<float>( in Execute() local
224 inputToOutputWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp155 std::unique_ptr<Decoder<float>> inputToOutputWeightsTensor = MakeDecoder<float>( in Execute() local
257 inputToOutputWeightsTensor, in Execute()
H A DLstm.hpp33 std::unique_ptr<Decoder<float>>& inputToOutputWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp114 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in QLstmEndToEnd() local
151 params.m_InputToOutputWeights = &inputToOutputWeightsTensor; in QLstmEndToEnd()
H A DQuantizedLstmEndToEndTestImpl.cpp72 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in CreateQuantizedLstmNetwork() local
107 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp536 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo4x5);
558 data.m_InputToOutputWeights = &inputToOutputWeightsTensor;