Searched defs: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
|
| H A D | UnidirectionalSequenceLstm.hpp | 195 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 3}, dataType, qScale, qOffset); in VisitUnidirectionalSequenceLstmOperator() local
|
| /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/backends/backendsCommon/test/layerTests/ |
| H A D | LstmTestImpl.cpp | 167 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
363 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1045 …armnn::TensorInfo scratchBufferTensorInfo({batchSize, scratchBufferSize}, ArmnnType, qScale, qOffs… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1270 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
|
| /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
|
| /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
|