| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | UnidirectionalSequenceLstmTestImpl.cpp | 40 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
61 … tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
76 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
236 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
257 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
272 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
425 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
446 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
461 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
684 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
|
| D | LstmTestImpl.cpp | 164 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
188 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
205 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
359 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
384 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
401 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1042 armnn::TensorInfo cellStateInTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1193 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1207 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1266 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | QuantizedLstmEndToEndTestImpl.cpp | 131 armnn::TensorInfo cellStateInTensorInfo({batchSize , outputSize}, in CreateQuantizedLstmNetwork() local
154 Connect(cellStateIn, quantizedLstmLayer, cellStateInTensorInfo, 0, 1); in CreateQuantizedLstmNetwork()
|
| D | WorkloadDataValidation.cpp | 507 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, dataType, qScale, qOffset); variable
526 AddInputToWorkload(data, info, cellStateInTensorInfo, nullptr);
|
| /external/armnn/delegate/test/ |
| D | LstmTest.cpp | 26 std::vector<int32_t> cellStateInTensorInfo {batchSize , numUnits}; in LstmTest() local
|
| D | UnidirectionalSequenceLstmTest.cpp | 170 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmTimeMajorTest() local
318 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
|