Searched refs:scratchBufferTensorInfo (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/armnn/delegate/classic/src/ |
H A D | Lstm.hpp | 162 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in VisitLstmOperator() local 165 … scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in VisitLstmOperator() 231 scratchBufferTensorInfo, in VisitLstmOperator() 249 layer->GetOutputSlot(0).SetTensorInfo(scratchBufferTensorInfo); in VisitLstmOperator()
|
H A D | UnidirectionalSequenceLstm.hpp | 195 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in VisitUnidirectionalSequenceLstmOperator() local 198 … scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in VisitUnidirectionalSequenceLstmOperator()
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | LstmTestImpl.cpp | 167 … 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 D | WorkloadDataValidation.cpp | 510 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 D | LstmTest.cpp | 29 std::vector<int32_t> scratchBufferTensorInfo {batchSize, numUnits * 4}; in LstmTest() local
|
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
H A D | TfLiteParser.cpp | 3722 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in ParseUnidirectionalSequenceLSTM() local 3725 … scratchBufferTensorInfo = armnn::TensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in ParseUnidirectionalSequenceLSTM()
|