| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | UnidirectionalSequenceLstmTestImpl.cpp | 43 …armnn::TensorInfo cellStateOutTensorInfo({ batchSize, 1, outputSize }, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
53 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
68 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
79 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
239 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, ArmnnType, qScale, qOf… in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
249 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
264 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
275 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
428 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
438 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
[all …]
|
| D | LstmTestImpl.cpp | 168 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
197 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
209 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
364 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
393 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
405 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1047 armnn::TensorInfo cellStateOutTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1174 LayerTestResult<T, 2> ret2(cellStateOutTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1184 std::vector<T> actualCellStateOutput(cellStateOutTensorInfo.GetNumElements()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1200 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
[all …]
|
| /external/armnn/delegate/classic/src/ |
| D | Lstm.hpp | 167 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, dataType, qScale, qOffset); in VisitLstmOperator() local
233 cellStateOutTensorInfo, in VisitLstmOperator()
251 layer->GetOutputSlot(2).SetTensorInfo(cellStateOutTensorInfo); in VisitLstmOperator()
|
| D | UnidirectionalSequenceLstm.hpp | 200 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, in VisitUnidirectionalSequenceLstmOperator() local
269 cellStateOutTensorInfo, in VisitUnidirectionalSequenceLstmOperator()
286 layer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo); in VisitUnidirectionalSequenceLstmOperator()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | QuantizedLstmEndToEndTestImpl.cpp | 141 armnn::TensorInfo cellStateOutTensorInfo({batchSize, outputSize}, in CreateQuantizedLstmNetwork() local
158 Connect(quantizedLstmLayer, cellStateOut, cellStateOutTensorInfo, 0, 0); in CreateQuantizedLstmNetwork()
|
| D | WorkloadDataValidation.cpp | 511 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, dataType, qScale, qOffset); variable
530 AddOutputToWorkload(data, info, cellStateOutTensorInfo, nullptr);
|
| /external/armnn/delegate/test/ |
| D | LstmTest.cpp | 30 std::vector<int32_t> cellStateOutTensorInfo {batchSize, numUnits}; in LstmTest() local
|
| /external/armnn/src/armnnSerializer/test/ |
| D | LstmSerializationTests.cpp | 2303 armnn::TensorInfo cellStateOutTensorInfo({ batchSize, outputSize }, armnn::DataType::Float32); variable
2316 unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2326 {outputStateOutTensorInfo, cellStateOutTensorInfo, outputTensorInfo},
2444 armnn::TensorInfo cellStateOutTensorInfo({ batchSize, outputSize }, armnn::DataType::Float32); variable
2457 unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2467 {outputStateOutTensorInfo, cellStateOutTensorInfo, outputTensorInfo},
2604 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… variable
2617 unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2627 {outputStateOutTensorInfo, cellStateOutTensorInfo, outputTensorInfo},
2713 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… variable
[all …]
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 3727 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, in ParseUnidirectionalSequenceLSTM() local
3797 layer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo); in ParseUnidirectionalSequenceLSTM()
|