Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
DUnidirectionalSequenceLstmTestImpl.cpp149 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
163 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
176 data.m_InputGateBias = &inputGateBiasTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
338 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
351 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
364 data.m_InputGateBias = &inputGateBiasTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
527 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
540 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
553 data.m_InputGateBias = &inputGateBiasTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
843 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo6); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
DLstmTestImpl.cpp279 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
293 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
306 data.m_InputGateBias = &inputGateBiasTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
939 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo20); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
957 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
975 data.m_InputGateBias = &inputGateBiasTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1399 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1422 AllocateAndCopyDataToITensorHandle(&inputGateBiasTensor, inputGateBias.data()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1445 data.m_InputGateBias = &inputGateBiasTensor; in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1618 armnn::ScopedTensorHandle inputGateBiasTensor(biasInfo); in QuantizedLstmTestImpl() local
[all …]
/external/armnn/src/backends/reference/workloads/
DLstm.cpp37 std::unique_ptr<Decoder<float>>& inputGateBiasTensor, in LstmImpl()
78 VectorBatchVectorAssign(*inputGateBiasTensor, in LstmImpl()
140 VectorBatchVectorAdd(*inputGateBiasTensor, in LstmImpl()
DRefLstmWorkload.cpp135 std::unique_ptr<Decoder<float>> inputGateBiasTensor; in Execute() local
177 inputGateBiasTensor = MakeDecoder<float>( in Execute()
232 inputGateBiasTensor, in Execute()
DRefUnidirectionalSequenceLstmWorkload.cpp166 std::unique_ptr<Decoder<float>> inputGateBiasTensor; in Execute() local
205 inputGateBiasTensor = MakeDecoder<float>( in Execute()
265 inputGateBiasTensor, in Execute()
DLstm.hpp41 std::unique_ptr<Decoder<float>>& inputGateBiasTensor,
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp93 armnn::ConstTensor inputGateBiasTensor(biasInfo, inputGateBiasTensorVector.data()); in CreateQuantizedLstmNetwork() local
112 data.m_InputGateBias = &inputGateBiasTensor; in CreateQuantizedLstmNetwork()
DWorkloadDataValidation.cpp542 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4);
564 data.m_InputGateBias = &inputGateBiasTensor;
600 data.m_InputGateBias = &inputGateBiasTensor;