Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
DUnidirectionalSequenceLstmTestImpl.cpp146 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
160 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
174 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
336 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
349 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
362 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
525 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
538 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
551 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
840 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo6x5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
DLstmTestImpl.cpp276 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
290 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
304 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
936 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
954 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
972 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1118 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1133 … AllocateAndCopyDataToITensorHandle(&recurrentToCellWeightsTensor, recurrentToCellWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1148 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1396 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/external/armnn/src/backends/reference/workloads/
DLstm.cpp32 std::unique_ptr<Decoder<float>>& recurrentToCellWeightsTensor, in LstmImpl() argument
121 MatrixBatchVectorMultiplyAccumulate(*recurrentToCellWeightsTensor, in LstmImpl()
DLstm.hpp36 std::unique_ptr<Decoder<float>>& recurrentToCellWeightsTensor,
DRefLstmWorkload.cpp130 std::unique_ptr<Decoder<float>> recurrentToCellWeightsTensor = MakeDecoder<float>( in Execute() local
227 recurrentToCellWeightsTensor, in Execute()
DRefUnidirectionalSequenceLstmWorkload.cpp161 std::unique_ptr<Decoder<float>> recurrentToCellWeightsTensor = MakeDecoder<float>( in Execute() local
260 recurrentToCellWeightsTensor, in Execute()
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp85 …armnn::ConstTensor recurrentToCellWeightsTensor(recurrentWeightsInfo, recurrentToCellWeightsTensor… in CreateQuantizedLstmNetwork() local
110 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in CreateQuantizedLstmNetwork()
DQLstmEndToEndTestImpl.cpp125 armnn::ConstTensor recurrentToCellWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local
154 params.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor; in QLstmEndToEnd()
DWorkloadDataValidation.cpp539 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo4x3);
561 data.m_RecurrentToCellWeights = &recurrentToCellWeightsTensor;