| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | UnidirectionalSequenceLstmTestImpl.cpp | 41 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
63 … tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
75 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
237 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
259 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
271 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
426 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
448 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
460 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
685 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
|
| D | LstmTestImpl.cpp | 165 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
190 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
204 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
360 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
386 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
400 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1041 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1191 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1206 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1267 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | QuantizedLstmEndToEndTestImpl.cpp | 136 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, in CreateQuantizedLstmNetwork() local
155 Connect(outputStateIn, quantizedLstmLayer, outputStateInTensorInfo, 0, 2); in CreateQuantizedLstmNetwork()
|
| D | WorkloadDataValidation.cpp | 506 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, dataType, qScale, qOffset); variable
525 AddInputToWorkload(data, info, outputStateInTensorInfo, nullptr);
|
| /external/armnn/delegate/test/ |
| D | LstmTest.cpp | 27 std::vector<int32_t> outputStateInTensorInfo {batchSize , outputSize}; in LstmTest() local
|
| D | UnidirectionalSequenceLstmTest.cpp | 171 std::vector<int32_t> outputStateInTensorInfo = {batchSize, outputSize}; in UnidirectionalSequenceLstmTimeMajorTest() local
319 std::vector<int32_t> outputStateInTensorInfo = {batchSize, outputSize}; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
|