Searched refs:outputTensorInfo4 (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnnUtils/test/ |
| H A D | ParserHelperTest.cpp | 57 TensorInfo outputTensorInfo4; variable
59 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData4, keepDims, outputTensorInfo4);
61 CHECK(outputTensorInfo4.GetNumDimensions() == 3);
62 CHECK(outputTensorInfo4.GetShape()[0] == 1);
63 CHECK(outputTensorInfo4.GetShape()[1] == 3);
64 CHECK(outputTensorInfo4.GetShape()[2] == 4);
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | SplitterTestImpl.cpp | 59 …armnn::TensorInfo outputTensorInfo4({ outputChannels1, outputHeight1, outputWidth1 }, ArmnnType, q… in SplitterTestCommon() local
73 outputTensorInfo4.SetQuantizationScale(qScale); in SplitterTestCommon()
74 outputTensorInfo4.SetQuantizationOffset(qOffset); in SplitterTestCommon()
160 std::vector<T> actualData4(outputTensorInfo4.GetNumElements()); in SplitterTestCommon()
197 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle2, outputTensorInfo4.GetShape(), wOrigin4.d… in SplitterTestCommon()
198 tensorHandleFactory.CreateTensorHandle(outputTensorInfo4); in SplitterTestCommon()
230 AddOutputToWorkload(data2, info2, outputTensorInfo4, outputHandle4.get()); in SplitterTestCommon()
250 …ult<T,3> ret4(actualData4, expectedData4, outputHandle4->GetShape(), outputTensorInfo4.GetShape()); in SplitterTestCommon()
|
| /aosp_15_r20/external/armnn/src/armnnDeserializer/ |
| H A D | Deserializer.cpp | 3333 armnn::TensorInfo outputTensorInfo4 = ToTensorInfo(outputs[3]); in ParseLstm() local
3334 layer->GetOutputSlot(3).SetTensorInfo(outputTensorInfo4); in ParseLstm()
|