Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp142 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
156 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
170 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
332 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
345 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
358 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
521 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
534 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
547 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
836 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
H A DLstmTestImpl.cpp272 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
286 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
300 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
932 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
950 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
968 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1110 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfoInput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1125 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1140 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1392 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp28 std::unique_ptr<Decoder<float>>& inputToCellWeightsTensor, in LstmImpl() argument
108 MatrixBatchVectorMultiplyAccumulate(*inputToCellWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp122 std::unique_ptr<Decoder<float>> inputToCellWeightsTensor = MakeDecoder<float>( in Execute() local
223 inputToCellWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp153 std::unique_ptr<Decoder<float>> inputToCellWeightsTensor = MakeDecoder<float>( in Execute() local
256 inputToCellWeightsTensor, in Execute()
H A DLstm.hpp32 std::unique_ptr<Decoder<float>>& inputToCellWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp113 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in QLstmEndToEnd() local
150 params.m_InputToCellWeights = &inputToCellWeightsTensor; in QLstmEndToEnd()
H A DQuantizedLstmEndToEndTestImpl.cpp69 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in CreateQuantizedLstmNetwork() local
106 data.m_InputToCellWeights = &inputToCellWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp535 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo4x5);
557 data.m_InputToCellWeights = &inputToCellWeightsTensor;