Lines Matching refs:tensorHandleFactory
124 const armnn::ITensorHandleFactory& tensorHandleFactory, in PermuteTensorData() argument
141 …std::unique_ptr<ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInf… in PermuteTensorData()
142 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in PermuteTensorData()
174 const armnn::ITensorHandleFactory& tensorHandleFactory, in PermuteInputsForConcat() argument
218 tensorHandleFactory, in PermuteInputsForConcat()
244 const armnn::ITensorHandleFactory& tensorHandleFactory, in PermuteOutputForConcat() argument
267 tensorHandleFactory, in PermuteOutputForConcat()
279 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concatenate() argument
318 tensorHandleFactory, in Concatenate()
332 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in Concatenate()
347 outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo); in Concatenate()
356 tensorHandleFactory.CreateSubTensorHandle(*outputHandle, in Concatenate()
359 … tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in Concatenate()
370 …std::unique_ptr<ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInf… in Concatenate()
405 tensorHandleFactory, in Concatenate()
425 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat1dTestImpl() argument
441 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat1dTestImpl()
463 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat2dTestImpl() argument
505 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat2dTestImpl()
521 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat2dDim0TestImpl() argument
528 workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 0, qScale, qOffset); in Concat2dDim0TestImpl()
559 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat2dDim1TestImpl() argument
566 workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 1, qScale, qOffset); in Concat2dDim1TestImpl()
585 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat2dDim0DiffInputDimsTestImpl() argument
627 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat2dDim0DiffInputDimsTestImpl()
665 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat2dDim1DiffInputDimsTestImpl() argument
707 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat2dDim1DiffInputDimsTestImpl()
733 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dTestImpl() argument
812 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat3dTestImpl()
828 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim0TestImpl() argument
835 … workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 0, true, qScale, qOffset); in Concat3dDim0TestImpl()
902 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim1TestImpl() argument
909 … workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 1, true, qScale, qOffset); in Concat3dDim1TestImpl()
976 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2TestImpl() argument
984 …workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 2, useSubtensor, qScale, qO… in Concat3dDim2TestImpl()
1015 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim0DiffInputDimsTestImpl() argument
1093 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat3dDim0DiffInputDimsTestImpl()
1167 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim1DiffInputDimsTestImpl() argument
1239 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat3dDim1DiffInputDimsTestImpl()
1307 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2DiffInputDimsTestImpl() argument
1386 Concatenate<T>(workloadFactory, memoryManager, tensorHandleFactory, in Concat3dDim2DiffInputDimsTestImpl()
1424 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dTestImpl() argument
1473 tensorHandleFactory, in Concat4dTestImpl()
1489 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDim0TestImpl() argument
1496 … workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 0, true, qScale, qOffset); in Concat4dDim0TestImpl()
1530 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDim1TestImpl() argument
1537 … workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 1, true, qScale, qOffset); in Concat4dDim1TestImpl()
1571 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDim2TestImpl() argument
1578 … workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 2, true, qScale, qOffset); in Concat4dDim2TestImpl()
1612 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDim3TestImpl() argument
1620 …workloadFactory, memoryManager, tensorHandleFactory, outputTensorInfo, 3, useSubtensor, qScale, qO… in Concat4dDim3TestImpl()
1654 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim0TestImpl() argument
1700 tensorHandleFactory, in Concat4dDiffShapeDim0TestImpl()
1741 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim1TestImpl() argument
1778 tensorHandleFactory, in Concat4dDiffShapeDim1TestImpl()
1809 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim2TestImpl() argument
1849 tensorHandleFactory, in Concat4dDiffShapeDim2TestImpl()
1887 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim3TestImpl() argument
1928 tensorHandleFactory, in Concat4dDiffShapeDim3TestImpl()
1955 const armnn::ITensorHandleFactory& tensorHandleFactory, in ConcatDifferentInputOutputQParamTest() argument
2045 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in ConcatDifferentInputOutputQParamTest()
2051 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo1.GetShape(), wOrigin1.dat… in ConcatDifferentInputOutputQParamTest()
2052 tensorHandleFactory.CreateTensorHandle(inputTensorInfo1); in ConcatDifferentInputOutputQParamTest()
2056 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo2.GetShape(), wOrigin2.dat… in ConcatDifferentInputOutputQParamTest()
2057 tensorHandleFactory.CreateTensorHandle(inputTensorInfo2); in ConcatDifferentInputOutputQParamTest()
2097 const armnn::ITensorHandleFactory& tensorHandleFactory,
2104 const armnn::ITensorHandleFactory& tensorHandleFactory,
2114 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatTest() argument
2197 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in ConcatTest()
2203 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo1.GetShape(), wOrigin1.dat… in ConcatTest()
2204 tensorHandleFactory.CreateTensorHandle(inputTensorInfo1); in ConcatTest()
2208 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo2.GetShape(), wOrigin2.dat… in ConcatTest()
2209 tensorHandleFactory.CreateTensorHandle(inputTensorInfo2); in ConcatTest()
2240 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat1dTest() argument
2242 …return Concat1dTestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.… in Concat1dTest()
2248 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim0Test() argument
2250 …cat2dDim0TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat2dDim0Test()
2256 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim1Test() argument
2258 …cat2dDim1TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat2dDim1Test()
2264 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim0DiffInputDimsTest() argument
2267 tensorHandleFactory, 0.0f, 0); in Concat2dDim0DiffInputDimsTest()
2273 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim1DiffInputDimsTest() argument
2277 tensorHandleFactory, in Concat2dDim1DiffInputDimsTest()
2285 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim0Test() argument
2287 …cat3dDim0TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat3dDim0Test()
2293 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim1Test() argument
2295 …cat3dDim1TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat3dDim1Test()
2301 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2Test() argument
2304 …return Concat3dDim2TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, in Concat3dDim2Test()
2311 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim0DiffInputDimsTest() argument
2314 workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat3dDim0DiffInputDimsTest()
2320 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim1DiffInputDimsTest() argument
2323 tensorHandleFactory, 0.0f, 0); in Concat3dDim1DiffInputDimsTest()
2329 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2DiffInputDimsTest() argument
2333 workloadFactory, memoryManager, tensorHandleFactory, useSubtensor, 0.0f, 0); in Concat3dDim2DiffInputDimsTest()
2339 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim0Test() argument
2341 …cat4dDim0TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat4dDim0Test()
2347 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim1Test() argument
2349 …cat4dDim1TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat4dDim1Test()
2355 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim2Test() argument
2357 …cat4dDim2TestImpl<DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat4dDim2Test()
2363 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDim3Test() argument
2367 tensorHandleFactory, 0.0f, 0, useSubtensor); in Concat4dDim3Test()
2373 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim0Test() argument
2376 tensorHandleFactory, 0.0f, 0); in Concat4dDiffShapeDim0Test()
2382 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim1Test() argument
2385 workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in Concat4dDiffShapeDim1Test()
2391 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim2Test() argument
2394 tensorHandleFactory, 0.0f, 0); in Concat4dDiffShapeDim2Test()
2400 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim3Test() argument
2404 workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0, useSubtensor); in Concat4dDiffShapeDim3Test()
2410 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatFloat16Test() argument
2412 …cat3dDim1TestImpl<DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in ConcatFloat16Test()
2418 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatBFloat16Test() argument
2420 …at3dDim1TestImpl<DataType::BFloat16>(workloadFactory, memoryManager, tensorHandleFactory, 0.0f, 0); in ConcatBFloat16Test()
2426 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatUint8DifferentQParamsTest() argument
2529 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in ConcatUint8DifferentQParamsTest()
2535 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo1.GetShape(), wOrigin1.dat… in ConcatUint8DifferentQParamsTest()
2536 tensorHandleFactory.CreateTensorHandle(inputTensorInfo1); in ConcatUint8DifferentQParamsTest()
2540 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo2.GetShape(), wOrigin2.dat… in ConcatUint8DifferentQParamsTest()
2541 tensorHandleFactory.CreateTensorHandle(inputTensorInfo2); in ConcatUint8DifferentQParamsTest()
2572 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatUint8Test() argument
2666 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in ConcatUint8Test()
2672 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo1.GetShape(), wOrigin1.dat… in ConcatUint8Test()
2673 tensorHandleFactory.CreateTensorHandle(inputTensorInfo1); in ConcatUint8Test()
2677 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo2.GetShape(), wOrigin2.dat… in ConcatUint8Test()
2678 tensorHandleFactory.CreateTensorHandle(inputTensorInfo2); in ConcatUint8Test()
2710 const armnn::ITensorHandleFactory& tensorHandleFactory) in ConcatUint16Test() argument
2802 …std::unique_ptr<ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorI… in ConcatUint16Test()
2808 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo1.GetShape(), wOrigin1.dat… in ConcatUint16Test()
2809 tensorHandleFactory.CreateTensorHandle(inputTensorInfo1); in ConcatUint16Test()
2813 …tensorHandleFactory.CreateSubTensorHandle(*outputHandle, inputTensorInfo2.GetShape(), wOrigin2.dat… in ConcatUint16Test()
2814 tensorHandleFactory.CreateTensorHandle(inputTensorInfo2); in ConcatUint16Test()
2846 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat1dUint8Test() argument
2848 …return Concat1dTestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0… in Concat1dUint8Test()
2854 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim0Uint8Test() argument
2856 …t2dDim0TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat2dDim0Uint8Test()
2862 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim1Uint8Test() argument
2864 …t2dDim1TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat2dDim1Uint8Test()
2870 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim0DiffInputDimsUint8Test() argument
2873 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat2dDim0DiffInputDimsUint8Test()
2879 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat2dDim1DiffInputDimsUint8Test() argument
2882 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat2dDim1DiffInputDimsUint8Test()
2888 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim0Uint8Test() argument
2890 …t3dDim0TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat3dDim0Uint8Test()
2896 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim1Uint8Test() argument
2898 …t3dDim1TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat3dDim1Uint8Test()
2904 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2Uint8Test() argument
2908 workloadFactory, memoryManager, tensorHandleFactory, useSubtensor, 0.5f, -1); in Concat3dDim2Uint8Test()
2914 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim0DiffInputDimsUint8Test() argument
2916 …t3dDim0TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat3dDim0DiffInputDimsUint8Test()
2922 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat3dDim1DiffInputDimsUint8Test() argument
2925 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat3dDim1DiffInputDimsUint8Test()
2931 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat3dDim2DiffInputDimsUint8Test() argument
2935 workloadFactory, memoryManager, tensorHandleFactory, useSubtensor, 0.5f, -1); in Concat3dDim2DiffInputDimsUint8Test()
2941 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim0Uint8Test() argument
2943 …t4dDim0TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDim0Uint8Test()
2949 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim1Uint8Test() argument
2951 …t4dDim1TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDim1Uint8Test()
2957 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDim2Uint8Test() argument
2959 …t4dDim2TestImpl<DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDim2Uint8Test()
2965 const armnn::ITensorHandleFactory& tensorHandleFactory, bool useSubtensor) in Concat4dDim3Uint8Test() argument
2968 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1, useSubtensor); in Concat4dDim3Uint8Test()
2974 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim0Uint8Test() argument
2977 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDiffShapeDim0Uint8Test()
2983 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim1Uint8Test() argument
2986 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDiffShapeDim1Uint8Test()
2992 const armnn::ITensorHandleFactory& tensorHandleFactory) in Concat4dDiffShapeDim2Uint8Test() argument
2995 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1); in Concat4dDiffShapeDim2Uint8Test()
3001 const armnn::ITensorHandleFactory& tensorHandleFactory, in Concat4dDiffShapeDim3Uint8Test() argument
3005 workloadFactory, memoryManager, tensorHandleFactory, 0.5f, -1, useSubtensor); in Concat4dDiffShapeDim3Uint8Test()