Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp279 armnn::ScopedCpuTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
293 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, &recurrentToOutputWeights[0][0… in LstmNoCifgNoPeepholeNoProjectionTestImpl()
307 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
981 armnn::ScopedCpuTensorHandle recurrentToOutputWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
999 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, &recurrentToOutputWeights[0][0… in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1017 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1155 armnn::ScopedCpuTensorHandle recurrentToOutputWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1171 …AllocateAndCopyDataToITensorHandle(&recurrentToOutputWeightsTensor, &recurrentToOutputWeights[0][0… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1187 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1465 armnn::ScopedCpuTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/external/armnn/src/armnn/layers/
DQuantizedLstmLayer.cpp244 ConstTensor recurrentToOutputWeightsTensor; in Accept() local
250 recurrentToOutputWeightsTensor = recurrentToOutputWeightsTensorCopy; in Accept()
251 inputParams.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in Accept()
DQLstmLayer.cpp375 ConstTensor recurrentToOutputWeightsTensor; in Accept() local
381 recurrentToOutputWeightsTensor = recurrentToOutputWeightsTensorCopy; in Accept()
382 inputParams.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in Accept()
DLstmLayer.cpp365 ConstTensor recurrentToOutputWeightsTensor; in Accept() local
371 recurrentToOutputWeightsTensor = recurrentToOutputWeightsTensorCopy; in Accept()
372 inputParams.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in Accept()
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp89 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local
111 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in CreateQuantizedLstmNetwork()
DQLstmEndToEndTestImpl.cpp123 armnn::ConstTensor recurrentToOutputWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local
151 params.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in QLstmEndToEnd()
DWorkloadDataValidation.cpp525 armnn::ScopedCpuTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3); in BOOST_AUTO_TEST_CASE() local
547 data.m_RecurrentToOutputWeights = &recurrentToOutputWeightsTensor; in BOOST_AUTO_TEST_CASE()
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp124 std::unique_ptr<Decoder<float>> recurrentToOutputWeightsTensor = MakeDecoder<float>( in Execute() local
247 MatrixBatchVectorMultiplyAccumulate(*recurrentToOutputWeightsTensor, in Execute()