Lines Matching refs:inputTensorInfo
27 armnn::TensorInfo& inputTensorInfo, in SpaceToDepthTestImpl() argument
40 inputTensorInfo = armnnUtils::Permuted(inputTensorInfo, NHWCToNCHW); in SpaceToDepthTestImpl()
44 armnnUtils::Permute(inputTensorInfo.GetShape(), NHWCToNCHW, in SpaceToDepthTestImpl()
56 inputTensorInfo.SetQuantizationScale(qScale); in SpaceToDepthTestImpl()
57 inputTensorInfo.SetQuantizationOffset(qOffset); in SpaceToDepthTestImpl()
66 …e_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SpaceToDepthTestImpl()
70 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in SpaceToDepthTestImpl()
112 armnn::TensorInfo inputTensorInfo; in SpaceToDepthSimpleTest1() local
119 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SpaceToDepthSimpleTest1()
124 inputTensorInfo, outputTensorInfo, input, outputExpected, desc); in SpaceToDepthSimpleTest1()
147 armnn::TensorInfo inputTensorInfo; in SpaceToDepthSimpleTest2() local
154 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SpaceToDepthSimpleTest2()
159 inputTensorInfo, outputTensorInfo, input, outputExpected, desc); in SpaceToDepthSimpleTest2()