Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp277 armnn::ScopedCpuTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
291 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, &recurrentToForgetWeights[0][0… in LstmNoCifgNoPeepholeNoProjectionTestImpl()
305 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
978 armnn::ScopedCpuTensorHandle recurrentToForgetWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
997 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, &recurrentToForgetWeights[0][0… in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1015 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1154 armnn::ScopedCpuTensorHandle recurrentToForgetWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1170 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, &recurrentToForgetWeights[0][0… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1186 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1462 armnn::ScopedCpuTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/external/armnn/src/armnn/layers/
DQuantizedLstmLayer.cpp224 ConstTensor recurrentToForgetWeightsTensor; in Accept() local
230 recurrentToForgetWeightsTensor = recurrentToForgetWeightsTensorCopy; in Accept()
231 inputParams.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in Accept()
DQLstmLayer.cpp355 ConstTensor recurrentToForgetWeightsTensor; in Accept() local
361 recurrentToForgetWeightsTensor = recurrentToForgetWeightsTensorCopy; in Accept()
362 inputParams.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in Accept()
DLstmLayer.cpp347 ConstTensor recurrentToForgetWeightsTensor; in Accept() local
353 recurrentToForgetWeightsTensor = recurrentToForgetWeightsTensorCopy; in Accept()
354 inputParams.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in Accept()
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp80 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local
109 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in CreateQuantizedLstmNetwork()
DQLstmEndToEndTestImpl.cpp119 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local
149 params.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in QLstmEndToEnd()
DWorkloadDataValidation.cpp522 armnn::ScopedCpuTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3); in BOOST_AUTO_TEST_CASE() local
545 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in BOOST_AUTO_TEST_CASE()
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp120 std::unique_ptr<Decoder<float>> recurrentToForgetWeightsTensor = MakeDecoder<float>( in Execute() local
243 MatrixBatchVectorMultiplyAccumulate(*recurrentToForgetWeightsTensor, in Execute()