Home
last modified time | relevance | path

Searched refs:outputHandle2 (Results 1 – 5 of 5) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DSplitterTestImpl.cpp185 std::unique_ptr<armnn::ITensorHandle> outputHandle2 = in SplitterTestCommon() local
192 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle2, outputTensorInfo3.GetShape(), wOrigin3.d… in SplitterTestCommon()
197 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle2, outputTensorInfo4.GetShape(), wOrigin4.d… in SplitterTestCommon()
205 AddOutputToWorkload(data, info, outputTensorInfo2, outputHandle2.get()); in SplitterTestCommon()
216 outputHandle2->Allocate(); in SplitterTestCommon()
223 CopyDataFromITensorHandle(actualData2.data(), outputHandle2.get()); in SplitterTestCommon()
228 AddInputToWorkload(data2, info2, outputTensorInfo2, outputHandle2.get()); in SplitterTestCommon()
248 …LayerTestResult<T,3> ret2(actualData2, expectedData2, outputHandle2->GetShape(), outputTensorInfo2… in SplitterTestCommon()
/external/armnn/src/backends/aclCommon/test/
DCreateWorkloadClNeon.hpp110 auto outputHandle2 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor2.m_Outputs[0]); in CreateMemCopyWorkloads() local
113 CHECK((outputHandle2->GetTensorInfo() == TensorInfo({2, 3}, DataType::Float32))); in CreateMemCopyWorkloads()
/external/armnn/src/backends/neon/test/
DNeonCreateWorkloadTests.cpp712 auto outputHandle2 = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Outputs[2]); variable
713 CHECK(TestNeonTensorHandleInfo(outputHandle2, TensorInfo({2, 7, 7}, DataType::Float32)));
/external/armnn/src/backends/reference/test/
DRefCreateWorkloadTests.cpp716 auto outputHandle2 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[2]); in RefCreateSplitterWorkloadTest() local
717 CHECK((outputHandle2->GetTensorInfo() == TensorInfo({ 2, 7, 7 }, DataType))); in RefCreateSplitterWorkloadTest()
/external/armnn/src/backends/cl/test/
DClCreateWorkloadTests.cpp817 auto outputHandle2 = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Outputs[2]); in ClSplitterWorkloadTest() local
818 predResult = CompareIClTensorHandleShape(outputHandle2, {2, 7, 7}); in ClSplitterWorkloadTest()