Lines Matching refs:inputTensorInfo
37 TensorInfo inputTensorInfo(inputShape, DataType, qScale, qOffset, true); in CreateSpaceToDepthNetwork() local
52 Connect(input, SpaceToDepth, inputTensorInfo, 0, 0); in CreateSpaceToDepthNetwork()
63 armnn::TensorInfo& inputTensorInfo, in SpaceToDepthEndToEnd() argument
73 PermuteTensorNhwcToNchw<float>(inputTensorInfo, inputData); in SpaceToDepthEndToEnd()
79 inputTensorInfo.GetShape(), in SpaceToDepthEndToEnd()
105 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest1() local
122 inputTensorInfo, in SpaceToDepthNhwcEndToEndTest1()
136 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNchwEndToEndTest1() local
153 inputTensorInfo, in SpaceToDepthNchwEndToEndTest1()
170 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest2() local
184 inputTensorInfo, in SpaceToDepthNhwcEndToEndTest2()
200 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNchwEndToEndTest2() local
216 inputTensorInfo, in SpaceToDepthNchwEndToEndTest2()