Home
last modified time | relevance | path

Searched refs:recurrentToInputWeightsTensor (Results 1 – 7 of 7) sorted by relevance

/external/armnn/src/armnn/layers/
DQuantizedLstmLayer.cpp214 ConstTensor recurrentToInputWeightsTensor; in Accept() local
220 recurrentToInputWeightsTensor = recurrentToInputWeightsTensorCopy; in Accept()
221 inputParams.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in Accept()
DQLstmLayer.cpp345 ConstTensor recurrentToInputWeightsTensor; in Accept() local
351 recurrentToInputWeightsTensor = recurrentToInputWeightsTensorCopy; in Accept()
352 inputParams.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in Accept()
DLstmLayer.cpp338 ConstTensor recurrentToInputWeightsTensor; in Accept() local
344 recurrentToInputWeightsTensor = recurrentToInputWeightsTensorCopy; in Accept()
345 inputParams.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in Accept()
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp119 std::unique_ptr<Decoder<float>> recurrentToInputWeightsTensor; in Execute() local
168 recurrentToInputWeightsTensor = MakeDecoder<float>( in Execute()
240 MatrixBatchVectorMultiplyAccumulate(*recurrentToInputWeightsTensor, in Execute()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp276 armnn::ScopedCpuTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
290 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, &recurrentToInputWeights[0][0]); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
304 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
979 armnn::ScopedCpuTensorHandle recurrentToInputWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
996 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, &recurrentToInputWeights[0][0]); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1014 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1463 armnn::ScopedCpuTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1485 …AllocateAndCopyDataToITensorHandle(&recurrentToInputWeightsTensor, &recurrentToInputWeights[0][0]); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1508 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1682 armnn::ScopedCpuTensorHandle recurrentToInputWeightsTensor(recurrentWeightsInfo); in QuantizedLstmTestImpl() local
[all …]
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp76 …armnn::ConstTensor recurrentToInputWeightsTensor(recurrentWeightsInfo, recurrentToInputWeightsTens… in CreateQuantizedLstmNetwork() local
108 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in CreateQuantizedLstmNetwork()
DWorkloadDataValidation.cpp523 armnn::ScopedCpuTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3); in BOOST_AUTO_TEST_CASE() local
544 data.m_RecurrentToInputWeights = &recurrentToInputWeightsTensor; in BOOST_AUTO_TEST_CASE()