Lines Matching refs:inputTensorInfo
74 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in SimplePooling3dTestImpl() local
83 inputTensorInfo.SetQuantizationScale(qScale); in SimplePooling3dTestImpl()
84 inputTensorInfo.SetQuantizationOffset(qOffset); in SimplePooling3dTestImpl()
92 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimplePooling3dTestImpl()
100 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in SimplePooling3dTestImpl()
107 result.m_Supported = handle.IsPooling3dSupported(inputTensorInfo, in SimplePooling3dTestImpl()
183 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( batchSize, channels, inputDepth, in… in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local
191 inputTensorInfo.SetQuantizationScale(qScale); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
192 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
257 PermuteNCDHWToNDHWC(input, inputTensorInfo); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
263 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
282 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleMaxPooling3dTestCommon() local
288 inputTensorInfo.SetQuantizationScale(qScale); in SimpleMaxPooling3dTestCommon()
289 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleMaxPooling3dTestCommon()
330 PermuteNCDHWToNDHWC(inputData, inputTensorInfo); in SimpleMaxPooling3dTestCommon()
336 inputData, outputData, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in SimpleMaxPooling3dTestCommon()
361 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 4, 4, 4 , dataLayout, ArmnnTy… in IgnorePaddingSimpleMaxPooling3dTestCommon() local
367 inputTensorInfo.SetQuantizationScale(qScale); in IgnorePaddingSimpleMaxPooling3dTestCommon()
368 inputTensorInfo.SetQuantizationOffset(qOffset); in IgnorePaddingSimpleMaxPooling3dTestCommon()
415 PermuteNCDHWToNDHWC(input, inputTensorInfo); in IgnorePaddingSimpleMaxPooling3dTestCommon()
421 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in IgnorePaddingSimpleMaxPooling3dTestCommon()
440 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleAveragePooling3dTestCommon() local
446 inputTensorInfo.SetQuantizationScale(qScale); in SimpleAveragePooling3dTestCommon()
447 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleAveragePooling3dTestCommon()
488 PermuteNCDHWToNDHWC(inputData, inputTensorInfo); in SimpleAveragePooling3dTestCommon()
494 inputData, outputData, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in SimpleAveragePooling3dTestCommon()
519 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 5, 3, 52, 60, 68, dataLayout, Armnn… in LargeTensorsAveragePooling3dTestCommon() local
525 inputTensorInfo.SetQuantizationScale(qScale); in LargeTensorsAveragePooling3dTestCommon()
526 inputTensorInfo.SetQuantizationOffset(qOffset); in LargeTensorsAveragePooling3dTestCommon()
533 for (unsigned int i = 0 ; i < inputTensorInfo.GetShape().GetNumElements(); ++i) in LargeTensorsAveragePooling3dTestCommon()
547 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in LargeTensorsAveragePooling3dTestCommon()
573 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo ( 1, 1, 4, 4, 4, dataLayout, ArmnnTy… in IgnorePaddingSimpleAveragePooling3dTestCommon() local
579 inputTensorInfo.SetQuantizationScale(qScale); in IgnorePaddingSimpleAveragePooling3dTestCommon()
580 inputTensorInfo.SetQuantizationOffset(qOffset); in IgnorePaddingSimpleAveragePooling3dTestCommon()
627 PermuteNCDHWToNDHWC(input, inputTensorInfo); in IgnorePaddingSimpleAveragePooling3dTestCommon()
633 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in IgnorePaddingSimpleAveragePooling3dTestCommon()
652 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleL2Pooling3dTestCommon() local
658 inputTensorInfo.SetQuantizationScale(qScale); in SimpleL2Pooling3dTestCommon()
659 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleL2Pooling3dTestCommon()
700 PermuteNCDHWToNDHWC(inputData, inputTensorInfo); in SimpleL2Pooling3dTestCommon()
706 inputData, outputData, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in SimpleL2Pooling3dTestCommon()
731 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in IgnorePaddingSimpleL2Pooling3dTestCommon() local
737 inputTensorInfo.SetQuantizationScale(qScale); in IgnorePaddingSimpleL2Pooling3dTestCommon()
738 inputTensorInfo.SetQuantizationOffset(qOffset); in IgnorePaddingSimpleL2Pooling3dTestCommon()
821 PermuteNCDHWToNDHWC(input, inputTensorInfo); in IgnorePaddingSimpleL2Pooling3dTestCommon()
827 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in IgnorePaddingSimpleL2Pooling3dTestCommon()
839 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTestCommon() local
868 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTestCommon()
874 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTestCommon()
886 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareMaxPooling3dTestCommon() local
915 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareMaxPooling3dTestCommon()
921 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareMaxPooling3dTestCommon()
933 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolTestCommon() local
962 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolTestCommon()
967 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolTestCommon()
979 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareAveragePooling3dTestCommon() local
1008 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareAveragePooling3dTestCommon()
1013 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareAveragePooling3dTestCommon()
1025 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolTestCommon() local
1054 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolTestCommon()
1060 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolTestCommon()
1072 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareL2Pooling3dTestCommon() local
1101 PermuteNCDHWToNDHWC<T>(input, inputTensorInfo); in AsymmetricNonSquareL2Pooling3dTestCommon()
1107 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in AsymmetricNonSquareL2Pooling3dTestCommon()
1141 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(batchSize, channelCount, inputDepth,… in ComparePooling3dTestCommon() local
1149 inputTensorInfo.SetQuantizationScale(qScale); in ComparePooling3dTestCommon()
1150 inputTensorInfo.SetQuantizationOffset(qOffset); in ComparePooling3dTestCommon()
1155 std::vector<T> input = MakeRandomTensor<T>(inputTensorInfo, 81715); in ComparePooling3dTestCommon()
1160 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ComparePooling3dTestCommon()
1165 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); in ComparePooling3dTestCommon()
1185 …<armnn::ITensorHandle> inputHandleRef = refTensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ComparePooling3dTestCommon()
1191 comparisonResult.m_Supported = handle.IsPooling3dSupported(inputTensorInfo, in ComparePooling3dTestCommon()
1202 SetWorkloadInput(refData, refInfo, 0, inputTensorInfo, inputHandleRef.get()); in ComparePooling3dTestCommon()