Searched refs:inputToCellWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | UnidirectionalSequenceLstmTestImpl.cpp | 142 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local 156 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 170 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() 332 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local 345 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl() 358 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl() 521 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo12); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local 534 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 547 data.m_InputToCellWeights = &inputToCellWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() 836 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo6x4); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local [all …]
|
H A D | LstmTestImpl.cpp | 272 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local 286 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl() 300 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl() 932 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo20x5); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local 950 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 968 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() 1110 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfoInput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local 1125 AllocateAndCopyDataToITensorHandle(&inputToCellWeightsTensor, inputToCellWeights.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1140 data.m_InputToCellWeights = &inputToCellWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() 1392 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo4x5); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local [all …]
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | Lstm.cpp | 28 std::unique_ptr<Decoder<float>>& inputToCellWeightsTensor, in LstmImpl() argument 108 MatrixBatchVectorMultiplyAccumulate(*inputToCellWeightsTensor, in LstmImpl()
|
H A D | RefLstmWorkload.cpp | 122 std::unique_ptr<Decoder<float>> inputToCellWeightsTensor = MakeDecoder<float>( in Execute() local 223 inputToCellWeightsTensor, in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 153 std::unique_ptr<Decoder<float>> inputToCellWeightsTensor = MakeDecoder<float>( in Execute() local 256 inputToCellWeightsTensor, in Execute()
|
H A D | Lstm.hpp | 32 std::unique_ptr<Decoder<float>>& inputToCellWeightsTensor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | QLstmEndToEndTestImpl.cpp | 113 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in QLstmEndToEnd() local 150 params.m_InputToCellWeights = &inputToCellWeightsTensor; in QLstmEndToEnd()
|
H A D | QuantizedLstmEndToEndTestImpl.cpp | 69 …armnn::ConstTensor inputToCellWeightsTensor(inputWeightsInfo, inputToCellWeightsTensorVector.data(… in CreateQuantizedLstmNetwork() local 106 data.m_InputToCellWeights = &inputToCellWeightsTensor; in CreateQuantizedLstmNetwork()
|
H A D | WorkloadDataValidation.cpp | 535 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo4x5); 557 data.m_InputToCellWeights = &inputToCellWeightsTensor;
|