Home
last modified time | relevance | path

Searched refs:inputWeightsInfo (Results 1 – 5 of 5) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp46 armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in CreateQuantizedLstmNetwork() local
63 … armnn::ConstTensor inputToInputWeightsTensor(inputWeightsInfo, inputToInputWeightsVector.data()); in CreateQuantizedLstmNetwork()
66 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in CreateQuantizedLstmNetwork()
69 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in CreateQuantizedLstmNetwork()
72 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in CreateQuantizedLstmNetwork()
DQLstmEndToEndTestImpl.cpp80 const armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in QLstmEndToEnd() local
112 …armnn::ConstTensor inputToForgetWeightsTensor(inputWeightsInfo, inputToForgetWeightsVector.data()); in QLstmEndToEnd()
113 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in QLstmEndToEnd()
114 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in QLstmEndToEnd()
/external/armnn/src/armnnSerializer/test/
DLstmSerializationTests.cpp1667 armnn::TensorInfo inputWeightsInfo({numUnits, inputSize}, variable
1681 …std::vector<int8_t> inputToForgetWeightsData = GenerateRandomData<int8_t>(inputWeightsInfo.GetNumE…
1682 …std::vector<int8_t> inputToCellWeightsData = GenerateRandomData<int8_t>(inputWeightsInfo.GetNumE…
1683 …std::vector<int8_t> inputToOutputWeightsData = GenerateRandomData<int8_t>(inputWeightsInfo.GetNumE…
1685 armnn::ConstTensor inputToForgetWeights(inputWeightsInfo, inputToForgetWeightsData);
1686 armnn::ConstTensor inputToCellWeights(inputWeightsInfo, inputToCellWeightsData);
1687 armnn::ConstTensor inputToOutputWeights(inputWeightsInfo, inputToOutputWeightsData);
1830 armnn::TensorInfo inputWeightsInfo({numUnits, inputSize}, variable
1855 …std::vector<int8_t> inputToForgetWeightsData = GenerateRandomData<int8_t>(inputWeightsInfo.GetNumE…
1856 …std::vector<int8_t> inputToCellWeightsData = GenerateRandomData<int8_t>(inputWeightsInfo.GetNumE…
[all …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp1575 armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in QuantizedLstmTestImpl() local
1608 armnn::ScopedTensorHandle inputToInputWeightsTensor(inputWeightsInfo); in QuantizedLstmTestImpl()
1609 armnn::ScopedTensorHandle inputToForgetWeightsTensor(inputWeightsInfo); in QuantizedLstmTestImpl()
1610 armnn::ScopedTensorHandle inputToCellWeightsTensor(inputWeightsInfo); in QuantizedLstmTestImpl()
1611 armnn::ScopedTensorHandle inputToOutputWeightsTensor(inputWeightsInfo); in QuantizedLstmTestImpl()
1790 armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in QLstmTestImpl() local
1828 armnn::ScopedTensorHandle inputToForgetWeightsTensor(inputWeightsInfo); in QLstmTestImpl()
1829 armnn::ScopedTensorHandle inputToCellWeightsTensor(inputWeightsInfo); in QLstmTestImpl()
1830 armnn::ScopedTensorHandle inputToOutputWeightsTensor(inputWeightsInfo); in QLstmTestImpl()
2028 armnn::TensorInfo inputWeightsInfo({numUnits, inputSize}, in QLstmTestImpl1() local
[all …]
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp851 armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in CreateQuantizedLstmWorkloadTest() local
868 std::make_unique<ScopedTensorHandle>(inputWeightsInfo); in CreateQuantizedLstmWorkloadTest()
870 std::make_unique<ScopedTensorHandle>(inputWeightsInfo); in CreateQuantizedLstmWorkloadTest()
872 std::make_unique<ScopedTensorHandle>(inputWeightsInfo); in CreateQuantizedLstmWorkloadTest()
874 std::make_unique<ScopedTensorHandle>(inputWeightsInfo); in CreateQuantizedLstmWorkloadTest()
949 CHECK((queueDescriptor.m_InputToInputWeights->GetTensorInfo() == inputWeightsInfo)); in CreateQuantizedLstmWorkloadTest()
950 CHECK((queueDescriptor.m_InputToForgetWeights->GetTensorInfo() == inputWeightsInfo)); in CreateQuantizedLstmWorkloadTest()
951 CHECK((queueDescriptor.m_InputToCellWeights->GetTensorInfo() == inputWeightsInfo)); in CreateQuantizedLstmWorkloadTest()
952 CHECK((queueDescriptor.m_InputToOutputWeights->GetTensorInfo() == inputWeightsInfo)); in CreateQuantizedLstmWorkloadTest()
1016 armnn::TensorInfo inputWeightsInfo({outputSize, inputSize}, in CreateQLstmWorkloadTest() local
[all …]