Home
last modified time | relevance | path

Searched refs:scratchBufferTensorInfo (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DLstm.hpp162 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in VisitLstmOperator() local
165scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in VisitLstmOperator()
231 scratchBufferTensorInfo, in VisitLstmOperator()
249 layer->GetOutputSlot(0).SetTensorInfo(scratchBufferTensorInfo); in VisitLstmOperator()
H A DUnidirectionalSequenceLstm.hpp195 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in VisitUnidirectionalSequenceLstmOperator() local
198scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in VisitUnidirectionalSequenceLstmOperator()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DLstmTestImpl.cpp167 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
193 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
207 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
363 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
389 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
403 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1045 …armnn::TensorInfo scratchBufferTensorInfo({batchSize, scratchBufferSize}, ArmnnType, qScale, qOffs… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1166 LayerTestResult<T, 2> ret0(scratchBufferTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1182 std::vector<T> actualScratchBufferOutput(scratchBufferTensorInfo.GetNumElements()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1196 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DWorkloadDataValidation.cpp510 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); variable
528 AddOutputToWorkload(data, info, scratchBufferTensorInfo, nullptr);
595 SetWorkloadOutput(data, info, 0, scratchBufferTensorInfo, nullptr);
/aosp_15_r20/external/armnn/delegate/test/
H A DLstmTest.cpp29 std::vector<int32_t> scratchBufferTensorInfo {batchSize, numUnits * 4}; in LstmTest() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3722 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in ParseUnidirectionalSequenceLSTM() local
3725scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in ParseUnidirectionalSequenceLSTM()