Searched refs:inputToOutputWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 143 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local 157 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 171 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 333 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local 346 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl() 359 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl() 522 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local 535 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 548 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 837 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local [all …]
|
H A D | LstmTestImpl.cpp | 273 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local 287 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl() 301 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl() 933 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local 951 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 969 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 1112 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfoInput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local 1127 AllocateAndCopyDataToITensorHandle(&inputToOutputWeightsTensor, inputToOutputWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1142 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1393 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local [all …]
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | Lstm.cpp | 29 std::unique_ptr<Decoder<float>>& inputToOutputWeightsTensor, in LstmImpl() argument 110 MatrixBatchVectorMultiplyAccumulate(*inputToOutputWeightsTensor, in LstmImpl()
|
H A D | RefLstmWorkload.cpp | 124 std::unique_ptr<Decoder<float>> inputToOutputWeightsTensor = MakeDecoder<float>( in Execute() local 224 inputToOutputWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 155 std::unique_ptr<Decoder<float>> inputToOutputWeightsTensor = MakeDecoder<float>( in Execute() local 257 inputToOutputWeightsTensor, in Execute()
|
H A D | Lstm.hpp | 33 std::unique_ptr<Decoder<float>>& inputToOutputWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QLstmEndToEndTestImpl.cpp | 114 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in QLstmEndToEnd() local 151 params.m_InputToOutputWeights = &inputToOutputWeightsTensor; in QLstmEndToEnd()
|
H A D | QuantizedLstmEndToEndTestImpl.cpp | 72 …armnn::ConstTensor inputToOutputWeightsTensor(inputWeightsInfo, inputToOutputWeightsTensorVector.d… in CreateQuantizedLstmNetwork() local 107 data.m_InputToOutputWeights = &inputToOutputWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 536 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo4x5); 558 data.m_InputToOutputWeights = &inputToOutputWeightsTensor;
|